记录在jetson tx2 nx上复现pvn3d过程中如何下载适配jetpack的cuda,torch,torchvision

        简单说一下这个问题,我个人觉得jetson的板子真的非常的抽象,版本固定太离谱了,一点扩展性没有,如果我一个项目有别的需求难不成再买一块你们的板子吗?但是没办法,人微言轻气抖冷,说一下结果。

        看起来是,jetson tx2 nx上只能安装4.x的jetpack,因为他只有ubuntu18版本,5.x系列都是ubuntu20版本,nx可以安装5.x,orin可以安装6.x,然后可以去官网看看,这些jetpack都支持一些什么版本的torch,并且每个版本的jetpack都有专用版本的cuda,联系更加紧密,不是说不能下载别的版本,只是可能会出现兼容问题。

        链接:PyTorch for Jetson - Announcements - NVIDIA Developer Forums

        这上面明确写了,什么版本的jetpack适配什么版本的torch,限制的死死的,我目前的jetson tx2 nx,只能安装4.x的jetpack所以就在图片中找,ok,如果你要安装原作者的torch1.5去复现,更抽象的事情来了。

         特么专用的sdkmanager不让你回溯到更早的版本,我看了一下,4.4的jetpack才可以安装torch1.5,我去问客服怎么办,他发了我一个长长的烧录方法,看着就复杂,直接给我劝退了,还需要在主机虚拟机上做,妈呀,真想骂人。

        没办法,找了一下,大家在torch1.8上也能实现,所以退一步,直接选择在1.8上实现,但是目前我没有找到,python3.6,torch1.8,cuda10.2并且还是arm架构的n卡上的复现博客,可以说是非常的抽象了。

        这是我参照的复现博客,大家也可以看一下,第一篇的话,是我在win10上复现的时候参照的https://blog.csdn.net/CHONGjjj/article/details/132077998,参照这篇,出了pcl部分有不一样的修改,后面说,其他的都是成功复现了。

第二篇https://blog.csdn.net/weixin_41837701/article/details/121000539,勉强参考一下,其实共同性不多。

安装torch和torchvision

        简单说一下怎么安装吧,之前图中的位置下载torch的whl文件,然后pip install就可以了。

        与之对应的torchvision在下面

下面有对应关系,然后参照上面的代码做就可以了,我帮你们翻译一下:

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev # 安装依赖
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # 把version改成v0.9.0即可
$ cd torchvision
$ export BUILD_VERSION=0.x.0  # 把x改成9  
$ python3 setup.py install --user
$ cd ../
$ pip install 'pillow<7' # 如果pillow版本较低,就这么做

cuda10.2应该不用多说,安装jetpack的时候自动安装了 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值