前言
昨天想在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就可以。
资源
因为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