在代码的开头，所有 import 之前，加入下面两行代码：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 这一行注释掉就是使用cpu，不注释就是使用gpu
TensorFlow进行GPU和CPU切换(强制使用CPU)
猜你喜欢
转载自blog.csdn.net/thisiszdy/article/details/89242019
今日推荐
周排行