Nvidia Xavier 上YOLOX demo运行环境

本文记录了作者在使用YOLOX模型在Jetson平台升级到CUDA 10.2和torch 1.7的过程中遇到的问题,包括pip安装错误、版本不匹配等,并分享了如何通过设置环境变量和选择合适版本的torchvision来解决问题的过程。
摘要由CSDN通过智能技术生成

虽然不喜欢csdn,不过为了让更多人看到,还是也记录在这里了。

由于YOLOX需要torch1.7以上运行,刷机了jetpack4.5, 更新到cuda10.2

过程一:

git下载YOLOX源码,运行官方command line

pip3 install -v -e .

报错(具体不记得了,应该是torchvision的问题,两个.o文件找不到)

然后我先做了一步 在bashrc中设置了 export OPENBLAS_CORETYPE=ARMV8

重新运行官方那个两个安装命令。在有报错的情况下安装完成yolox,可以用pip list 看见这个库

过程二:

在运行demo验证的时候发现报错,网上说的pytorch 和torchvision 版本的对应建议采用以下文中说明

PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums

本人安装的1.7 torch 对应版本的torchvision 应该是0.8.1(其他网上说对应的是0.8.0也可以然后我报错版本不兼容)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值