仅在 CPU 上运行 TensorFlow - 使用 CUDA VISIBLE DEVICES 环境变量
要确保 GPU 版 TensorFlow 进程仅在 CPU 上运行:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import tensorflow as tf
要确保 GPU 版 TensorFlow 进程仅在 CPU 上运行:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import tensorflow as tf