python - 阻止 TensorFlow 访问 GPU？

Question

有没有办法完全在 CPU 上运行 TensorFlow。我机器上的所有内存都被一个运行 TensorFlow 的单独进程占用。我尝试将 per_process_memory_fraction 设置为 0，但未成功。

Answer 1

看看这个question或者这个 answer .

总结一下，你可以添加这段代码:

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

使用 CUDA_VISIBLE_DEVICES 环境变量是 if not the way 之一每当您安装了 GPU-tensorflow 并且您不想使用任何 GPU 时都可以使用。

You to want either export CUDA_VISIBLE_DEVICES= or alternatively use a virtualenv with a non-GPU installation of TensorFlow.