Firefly开源社区

标题: RK3588S-PC-V1.0关于pytorch版本问题咨询 [打印本页]

作者: Q.IAN 时间: 2022-6-15 11:50
标题: RK3588S-PC-V1.0关于pytorch版本问题咨询
您好：
我现在购买的板子是RK3588S-PC-V1.0的版本，根据官方教程，安装的api是基于cp38的-1.3.0的，requirements里面记录的torch==1.6.0，torchvision==0.7.0。我在运行官方example时候，是没有问题的。
我想咨询下，只支持torch==1.6.0的版本吗？高于这个版本的，还能支持吗？谢谢
下面是我基于yolov5官方代码训练的模型，在板子上转模型时候，遇到的问题。如下所示：

我现在升级板子的torch版本到1.7.0，与我目前YOLOv5要求的环境一致，就提示了这个错误，意思是不能升级吗？

我在基于torch==1.7.0的环境进行同样的转模型操作，报如下的错误。转模型的相关代码和文件，麻烦给解答下，再次感谢

官方回复：requirements里面记录的版本肯定是支持的。高于或者低于requirements的版本即使可以转换成功，也有可能遇到一些奇奇怪怪的问题。所以目前最好使用requirements记录的版本

所以，想问问，关于这个版本问题，大家有没有遇到类似的情况，谢谢

tchip_askquestions

log.txt

6 KB, 下载次数: 0, 下载积分: 灯泡 -1 , 经验 -1

欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/)