【报错】raise pickle.UnpicklingError(_get_wo_message(str(e))) from None_pickle.UnpicklingError: Weigh

目录

报错

分析

解决方法

1 直接方法

2 推荐方法

3 最安全的方法

报错

解决方法

报错

使用rocm/pytorch:latest官方镜像，启动容器并运行模型推理。报错如下：

torch.load(str("./checkpoint/model.pt"), map_location=device)['model'].float().fuse().eval()

  File "/opt/conda/envs/py_3.12/lib/python3.12/site-packages/torch/serialization.py", line 1470, in load   raise pickle.UnpicklingError(_get_wo_message(str(e))) from None