cvd = os.environ['CUDA_VISIBLE_DEVICES'].strip()
File "/usr/lib/python2.7/UserDict.py", line 23, in __getitem__
raise KeyError(key)
KeyError: 'CUDA_VISIBLE_DEVICES'
如果没有,自己export一个就行了
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6
cvd = os.environ['CUDA_VISIBLE_DEVICES'].strip()
File "/usr/lib/python2.7/UserDict.py", line 23, in __getitem__
raise KeyError(key)
KeyError: 'CUDA_VISIBLE_DEVICES'
如果没有,自己export一个就行了
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6