2016-11-21 154 views
3

當我運行我的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

可能有人請解釋什麼必須繼續?

回答

7

援引CUDA_VISIBLE_DEVICES的解釋:

CUDA將枚舉可見的設備從零開始。在過去的 情況下,設備0,2,3會所以,如果你CUDA_VISIBLE_DEVICES=2,那麼你的GPU#2將被標註爲gpu:0內tensorflow顯示爲裝置0,1,2