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
。