Jetson Xavier NX在虚拟环境中配置pytorch

本篇主要参考下面文章，纪录一下自己的错误
1 NVIDIA Jetson NX配置深度学习环境（Pytorch、torchvision）超级详细，torchvision安装避坑总结
2 解决英伟达Jetson平台使用Python时的出现“Illegal instruction(cpre dumped)”错误

准备python环境

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install git cmake python3-dev
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev

 sudo apt-get install python3-pip
 sudo pip3 install -U pip testresources setuptools

下载pytorch编译好的.whl文件

注意现在还没有python3.7的.whl文件
我一开始下载了torch-1.5.0-cp36-cp36m-linux_aarch64.whl ，然后安装torchvision的时候出现了下面的错

我没有解决，重新下载了torch-1.6.0-cp36-cp36m-linux_aarch64.whl

创建虚拟环境

which python3.6 #查看python3.6路径
mkvirtualenv -p /usr/bin/python3.6 torch16

安装 pytorch

sudo apt-get install libopenblas-base libopenmpi-dev

不能用sudo安装mpi4py和Cython，它不是虚拟环境中的python版本

sudo pip3 install mpi4py
sudo pip3 install Cython

cd进存放.whl的文件目录

不能用sudo pip install torch-1.6.0-cp36-cp36m-linux_aarch64.whl

Error：torch-1.6.0-cp36-cp36m-linux_aarch64.whl is not a supported wheel on this platform

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

安装torchvision

进入其官方GitHub：https://github.com/pytorch/vision/tree/master
找到自己需要的版本下载下来

在终端中进vision目录

python3 setup.py install

先出现了“Illegal instruction(cpre dumped)”错误
解决方法

将“export OPENBLAS_CORETYPE=ARMV8”加入到“~/.bashrc”中

出现了参考博客的中的错

同样的解决方法

sudo apt install libavcodec-dev
sudo apt install libavformat-dev
sudo apt install libswscale-dev

成功

验证torch