【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之后,又能训练了!
目前还没查到真正的原因,只能说尽量在同一系列的显卡上训练和预测模型。
如果后期有查到原因会及时更新。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值