解决在cuda上安装torch后torch.cuda.is_available()返回False

问题： raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

主要原因：torch下载的是cpu版本的

image.png

解决办法：重新安装torch_cuda版本

解决步骤：
1.查看是否安装cuda；在cmd下输入 nvcc -V 查看版本，确定已经安装cuda及版本

image.png

2. conda list 查看依赖的包，比如python版本

3.到 https://download.pytorch.org/whl/torch_stable.html 找到 对应的cuda版本及python的轮子

image.png

4.复制链接，在conda下安装 torchvision 和 torch；安装torchvision 是确保版本一致，不过可以先安装torch

image.png

image.png