仅在 CPU 上运行 TensorFlow - 使用 CUDA VISIBLE DEVICES 环境变量

要确保 GPU 版 TensorFlow 进程仅在 CPU 上运行:

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"    
import tensorflow as tf

有关 CUDA_VISIBLE_DEVICES 的更多信息,请查看此答案CUDA 文档