Python KeyError:';CUDA可视设备&x27;

Python KeyError:';CUDA可视设备&x27;,python,pytorch,Python,Pytorch,今天我问了关于PyTorch在我的GPU上运行的问题。作为回应,有人建议运行以下代码: import os num_gpus = os.environ['CUDA_VISIBLE_DEVICES'].split(',').__len__() os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(f'{i}' for i in range(num_gpus)) 但是,在运行第二行之后,不幸的是,我遇到了另一个错误: num_gpus = os.environ

今天我问了关于PyTorch在我的GPU上运行的问题。作为回应,有人建议运行以下代码:

import os
num_gpus = os.environ['CUDA_VISIBLE_DEVICES'].split(',').__len__()
os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(f'{i}' for i in range(num_gpus))
但是,在运行第二行之后,不幸的是,我遇到了另一个错误:

num_gpus = os.environ['CUDA_VISIBLE_DEVICES'].split(',').__len__()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/hbp/anaconda3/lib/python3.8/os.py", line 675, in __getitem__
    raise KeyError(key) from None
KeyError: 'CUDA_VISIBLE_DEVICES'
num_gpus=os.environ['CUDA_VISIBLE_DEVICES'].split(',')。uu len_u()
回溯(最近一次呼叫最后一次):
文件“”,第1行,在
文件“/home/hbp/anaconda3/lib/python3.8/os.py”,第675行,在__
从“无”提升钥匙错误(钥匙)
KeyError:“CUDA\u可见\u设备”
我试着去查,但找不到任何有用的东西


有人知道我可以尝试什么吗?我链接到的问题可能有相关信息。

很明显,CUDA\u VISIBLE\u设备未设置且未引发问题默认情况下
CUDA\u VISIBLE\u设备不存在。当你的机器只有一个GPU时,做一些花哨的事情是没有意义的。只需使用
os.environ['CUDA\u VISIBLE\u DEVICES']=“0”
。感谢您的帮助!仅供参考,这不起作用,但我已经调整,所以我不再需要这个。