pytorch遇到一个问题

进入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”，就好了！