当尝试运行Python脚本或Jupyter Notebook时，出现'No CUDA GPUs are available'错误，尽管torch.cuda.is_available()返回True。问题可能在于环境变量CUDA_VISIBLE_DEVICES的设置，该变量被设置为非存在的GPU ID。将'2'或'1'改为'0'后，错误消失，表明代码中的CUDA设备选择出现了问题。
进入python时，
import torch
print(torch.cuda.device_count())
torch.cuda.is_available()
结果是：
1
True
而且根据网上方法检验了cuda和cudnn的安装都没有问题。但是但我运行python脚本或者在jupyter notebook中都会报错：
RuntimeError：No CUDA GPUs are available
注意此时：可能不是环境和配置的问题，而是代码里面出了问题，如果你的代码里面有：
os.environ["CUDA_VISIBLE_DEVICES"]="2"
并且在torch.cuda.is_available()的前面，那就是这里除了问题！
把“2”或者“1”改成“0”，就好了！
扫一扫
8万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
