全局配置
运行TensorFlow代码时候常出现OOM(Out of Memory)的错误，原因是batch_size设置得太大导致显存不足。如果想让代码仅仅运行在CPU下，可在原代码中加入如下代码：
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
Tensorflow 2.0/Keras 指定CPU运行,强制使用CPU
猜你喜欢
转载自blog.csdn.net/MrCharles/article/details/103708297
今日推荐
周排行