在进行实验的时候,需要使用到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