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之前設置變量（通常是在啟動腳本之前）。