Tensorflow:CUDA_VISIBLE_DEVICES似乎并没有工作

问题描述:

当我运行我的python脚本与CUDA_VISIBLE_DEVICES=2,Tensorflow还显示以下内容:Tensorflow:CUDA_VISIBLE_DEVICES似乎并没有工作

我tensorflow /核心/ common_runtime/GPU/gpu_device.cc:1041]创建 TensorFlow设备(/ GPU:0) - >(设备:0,名称:特斯拉K80,PCI总线ID: 0000:86:00.0)

因此,我的代码失败,出现以下消息:

无法满足显式设备规范'/ device:GPU'2' ,因为在此 进程中未注册符合该规范的设备;可用设备:/职务:本地主机/副本:0 /任务:0/CPU:0, /职业:本地主机/副本:0 /任务:0/GPU:0

可能有人请解释什么必须继续?

援引CUDA_VISIBLE_DEVICES的解释:

CUDA将枚举可见的设备从零开始。在过去的 情况下,设备0,2,3会所以,如果你CUDA_VISIBLE_DEVICES=2,那么你的GPU#2将被标注为gpu:0内tensorflow显示为装置0,1,2