问题:
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.
就我目前遇到的情况来说,如果你下载的cpu版本的torch,然后去torch.load(“一个模型”),这时候,我觉得很有可能会遇到这种情况。
因为你的device是cpu,但是如果你torch.load(“一个模型”)这里不固定一下参数,就报错了。
以我的代码为例,我环境,当时由于是用的清华源下载的,下载成cpu版本的了。
因此,当你用到torch.load的时候,没有指定的话,我觉得就会报错。
解决办法,可以跟我上面一样,在torch.load中指定一下cpu。