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

时间:2018-05-10 22:49:20

标签: python-3.x tensorflow

如何在我的CPU和GPU上快速切换运行张量流代码?

我的设置:

OS = 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如何运行代码。

0 个答案:

没有答案