问题: 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版本的
解决办法:重新安装torch_cuda版本
解决步骤:
1.查看是否安装cuda;在cmd下输入 nvcc -V 查看版本,确定已经安装cuda及版本
- conda list 查看依赖的包,比如python版本
3.到 https://download.pytorch.org/whl/torch_stable.html 找到 对应的cuda版本及python的轮子
4.复制链接,在conda下安装 torchvision 和 torch;安装torchvision 是确保版本一致,不过可以先安装torch
image.png image.png
网友评论