RuntimeError:No CUDA GPUs are available解决

进入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”,就好了!

你可能感兴趣的:(pytorch,神经网络)