Windows上的tf.device()故障

时间:2017-03-09 00:38:03

标签: python tensorflow

tf.device('/ gpu:0')无法仅在gpu0上运行模型,其他gpus也将被占用。结果,我服务器上所有gpu卡的所有内存都被填满。通过设置'config.gpu_options.visible_device_list =“0”',模型将仅在gpu0上运行。

BTW,只填充其他gpus的内存,不对它们进行实际计算,所有计算都在gpu0上执行。

任何能告诉我发生了什么事的人都非常感激。

环境信息

操作系统:Windows Server 2012 R2 CUDA:8.0 没有cuDNN

您安装的pip包的链接:

使用'pip install tensorflow-gpu'进行安装

python -c的输出“import tensorflow; print(tensorflow .__ version __)”。

1.0.1

您尝试了哪些其他尝试的解决方案?

config = tf.ConfigProto()

config.gpu_options.visible_device_list ='0'

使用'config'打开会话,该图只能在'/ gpu:0'上运行;

0 个答案:

没有答案