【Libtorch】报错:PytorchStreamReader failed reading zip archive: failed finding central directory

如题,本人出错的情况是在一台RTX3060显卡的机器上使用Libtorch训练的模型放在一台RTX3070显卡的机器上,但是居然发生了如上的错误。

出错的函数是torch::load(your_model.pt);

但是本人查阅了不少资料后发现都是版本的问题,但是本人使用的两台机器上的Libtorch都是1.7.1版本,所以这点被排除了。
参考资料:
【debug】PytorchStreamReader failed reading zip archive: failed finding central directory

然后本人试了下在3070显卡的机器上训练出一个模型,结果能正常推理预测了。但是本人目前不是很了解原因。望大佬可以指点下,感激不尽。

所以结论是:最好是使用什么型号的显卡来生成pt模型,就使用什么型号的型号的显卡来进行推理。

更新2022.10.26:
最近在本人的一台服务器上把一块RTX 1080Ti换成了RTX 3060后,Libtorch无法训练模型!
仔细检查问题点,发现是将模型设置到CUDA上这一步一直卡着了
然后我把3060换成1080Ti之后,又能训练了!
目前还没查到真正的原因,只能说尽量在同一系列的显卡上训练和预测模型。
如果后期有查到原因会及时更新。

你可能感兴趣的:(Libtorch,深度学习,深度学习,人工智能,pytorch)