【资源分享】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

你可能感兴趣的:(yolo,python,pytorch,深度学习,jetson,nx)