资源分享jetson xavier nx（aarch64）平台上的pytorch1.8.0+torchvision0.9.0（python=3.6）

Posted 2021-10-24 Rainbowman 0

前言

昨天想在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