TensorFlow只在CPU上运行的方法

import os
os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID”
os.environ[“CUDA_VISIBLE_DEVICES”] = “-1”

猜你喜欢

转载自blog.csdn.net/intjun/article/details/81479194