DNN_model.load_state_dict(torch.load('DNN_model.pth'))代码在这一步报错,错误为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.
时间: 2024-03-25 15:36:38 浏览: 152
这个错误的意思是你的代码在尝试将模型加载到 CUDA 设备上,但是你的电脑上没有可用的 CUDA 设备。建议你使用`torch.load`,并设置`map_location=torch.device('cpu')`来将模型加载到 CPU 上。可以尝试如下代码:
```python
device = torch.device('cpu') # 加载到CPU上
model = DNN_model().to(device) # 模型也要移到CPU上
model.load_state_dict(torch.load('DNN_model.pth', map_location=device))
```
这样,你就可以将模型加载到 CPU 上,并解决这个错误。
阅读全文