KeyError: 'CUDA_VISIBLE_DEVICES'

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

你可能感兴趣的:(KeyError: 'CUDA_VISIBLE_DEVICES')