解决RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is F

报错问题：

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.

解决方法：

这个错误通常是由于在 CPU 机器上尝试加载在 GPU 上保存的 PyTorch 模型导致的。

为了解决这个问题,你需修改你的代码如下:要在加载模型时指定 map_location 参数,将模型映射到 CPU 设备上。

修改你的代码如下：

import torch  

# 加载模型  
model = torch.load('path/to/model.pth', map_location=torch.device('cpu'))