Tensorflow：在CPU和GPU之间切换[Windows 10]

Question

如何在使用CPU和GPU运行Tensorflow代码之间快速切换？

我的设置：

操作系统= Windows 10

Python = 3.5

Tensorflow-gpu = 1.0.0

CUDA = 8.0.61

cuDNN = 5.1

我看到一则帖子建议设置CUDA_VISIBLE_DEVICES=0但我的环境中没有此变量（不确定是因为我正在运行Windows还是什么），但是如果我使用os.environ之类的方法进行设置，则不会不会影响tensorflow如何运行代码。

Answer 1

如果设置环境变量CUDA_VISIBLE_DEVICES=-1 ，则将仅使用CPU。 如果您未设置该环境变量，则将为所有GPU分配内存，但默认情况下仅使用GPU0。您也可以将其设置为要使用的特定GPU。 CUDA_VISIBLE_DEVICES=0将仅使用GPU 0。

此环境变量由用户创建，直到您创建它才存在。 您需要在导入tensorflow之前设置变量（通常是在启动脚本之前）。