僅在 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