【资源分享】jetson xavier nx(aarch64)平台上的pytorch1.8.0+torchvision0.9.0(python=3.6)

前言

昨天想在jetson xavier nx上跑下yolov5,就Git了官方的代码然后开始配置环境,一开始直接

pip install -r requirements.txt

倒是可以配置好环境,但跑demo时发现帧率不到1fps,用

sudo jtop

看了下资源使用情况,发现CPU拉满了,GPU还基本是0。
调查了一番发现是因为直接用requirements.txt配置环境的话,下载的pytorch是最新的1.9.0,对应的torchvision是0.10.0,和jetson xavier nx的cuda版本(cuda10.2)不对应。所以没法用cuda,也就没法使用GPU。

但现在有个问题,yolov5要求pytorch版本≥1.7.0,但我的pytorch是1.6.0(torchvision是0.7.0),不符合要求。所以决定升级pytorch。

但一定要注意匹配自己的cuda版本,因为我是cuda10.2所以pytorch1.8.0就可以。

cuda和pytorch对应版本

资源

因为jetson xavier nx是arm架构,和86架构不一样,所以不论是pypi上还是镜像上一些库的资源都不全,比如pytorch就只有1.9.0和很老的版本,没有1.8.0,torchvision也类似。这里放下已经编译好的可以在jetson xavier nx上安装的pytorch和torchvision。(python版本为3.6)

百度链接:https://pan.baidu.com/s/1LCTBv9zQj3ERLx7InY-sOg
提取码:1111

资源来源:https://download.csdn.net/download/qq_51095731/16753369

  • 4
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

SinHao22

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值