标签: 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如何运行代码。
CUDA_VISIBLE_DEVICES=0
os.environ