栏目分类:
子分类:
返回
终身学习网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
终身学习网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间:发布时间: 百科书网 趣学号
前言

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

转载请注明:文章转载自 www.051e.com
本文地址:http://www.051e.com/it/295776.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 ©2023-2025 051e.com

ICP备案号:京ICP备12030808号