僅在 CPU 上執行 TensorFlow Graph - 使用 tf.config

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(device_count={'GPU': 0}))

請記住，此方法會阻止 TensorFlow 圖使用 GPU，但 TensorFlow 仍會鎖定 GPU 裝置，如此方法中開啟的問題所述。使用 CUDA_VISIBLE_DEVICES 似乎是確保 TensorFlow 遠離 GPU 卡的最佳方法（請參閱此答案 ）。