問(wèn)題： 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 查看版本尝蠕，確定已經(jīng)安裝cuda及版本
image.png
- conda list 查看依賴(lài)的包，比如python版本
3.到 https://download.pytorch.org/whl/torch_stable.html 找到 對(duì)應(yīng)的cuda版本及python的輪子
image.png
4.復(fù)制鏈接早处，在conda下安裝 torchvision 和 torch氯哮；安裝torchvision 是確保版本一致，不過(guò)可以先安裝torch
image.png
image.png
