RuntimeError: Attempting to deserialize object on a CUDA device but torch.cu
时间: 2023-11-13 18:53:00 浏览: 174
runtimeerror_notifier:runtimeerror_notifier gem 适用于那些没有电子邮件发送资源的人
RuntimeError:尝试在CUDA设备上反序列化对象,但torch.cuda.is_available()为False。如果您正在运行的是仅CPU的机器,请使用torch.load并设置map_location='cpu'将存储映射到CPU。这个错误通常发生在尝试在没有可用的CUDA设备的情况下加载已经在CUDA设备上训练过的模型。解决这个问题的方法是使用map_location参数将模型加载到CPU上。
阅读全文