jetson tx2安装pytorch

在进行实验的时候,需要使用到pytorch,因此安装了一个archiconda,在conda中安装pytorch还是很简单,直接在https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048中下载wheel,使用pip install安装即可。

但是却有一个问题,就是需要使用opencv的时候,我同样pip install opencv-python,这时就会安装上最新版本的opencv-python,但是这会导致环境中的numpy被更新到1.19.5,这样就会导致倒入pytorch错误,因为jetson tx2要求使用1.13.3版本的numpy。

于是我pip uninstall numpy, pip install numpy==1.13.3,但是这样又会导致import cv2的时候报错。

我就放弃了archiconda的方案,使用系统自带的python3.6.9环境,发现其中自带opencv-python,因此现在就需要安装pytorch。

打开jtop观察到jetpack版本为4.6,因此我下载了torch-1.10.0-cp36-cp36m-linux_aarch64.whl ,使用

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

发现提示torch-1.10.0-cp36-cp36m-linux_aarch64.whl is not a supported wheel on this platform.

我尝试使用pip3来安装,但是发现系统没有安装pip3,于是我使用

sudo apt-get install python3-pip

来安装pip3,安装成功。

输入代码即可完成torch安装

pip3 install torch-1.10.0-cp36-cp36m-linux_aarch64.whl 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

zzj350123

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值