Firefly开源社区

打印 上一主题 下一主题

[技术讨论] RK3588S-PC-V1.0关于pytorch版本问题咨询

11

积分

0

威望

0

贡献

技术小白

积分
11

RK3588S-PC-V1.0关于pytorch版本问题咨询

发表于 2022-6-15 11:50:58      浏览:5028 | 回复:0        打印      只看该作者   [复制链接] 楼主
问题描述及复现步骤:
您好:
     我现在购买的板子是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记录的版本


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


log.txt

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

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表