解决在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

  1. 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
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容