TensorFlow只用CPU

转自 https://www.cnblogs.com/excellent-ship/p/8604941.html

限制TensorFlow只在CPU上运行的方法

import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

注：上述代码一定要放在import tensorflow或keras等之前，否则不起作用。