jetson Xavier nx安装torch和torchvision,并解决解决版本不匹配(报错RuntimeError: Couldn‘t load custom C++ ops)的问题

   

目录

1、安装torch

2、安装torchvision

3、验证是否安装成功

4、 错误记录

5、torch和torchvision网盘链接


首先,torch和torchvision都不能直接pip安装,

以下的演示是torch1.10.0  +    torchvision v0.11.1

1、安装torch

torch要去英伟达官网上找。链接如下(科学上网)https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048,(百度网盘里也有,后边在写)

上边网址打开之后是这样

 根据你的cuda版本(查询cuda版本,命令用nvcc -v),我是cuda10.2,按如下对应关系,

 下载torch版本,挑一个下载就行,下载完的文件像这样,这三个文件的百度网盘链接在最后

 下载完后用u盘转到jetson nx 板子上,主目录下新建个文件夹放好(位置没有要求,这样做就是为了删除的时候好找)

在板子上,进入放torch的文件夹,打开终端,进入你的conda虚机环境(如果你有的话)

运行

pip3 install torch-1.10.0-cp36-cp36m-linux_aarch64.whl

 注意torch文件名换成自己的,安装完成后,会提示successful install torch1.10.0,就安装成功了

2、安装torch

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值