Jetson Xavier NX部署YOLOV8+跟踪+识别算法2——conda、torch与torchvision的安装

前言：虽然Jetson Xavier NX是Linux系统，但是由于Jetson Xavier NX是ARM架构的，很多python库的安装过程要略微难于普通的Linux服务器。具体来说是Conda环境需要使用Archiconda而不是anaconda \miniconda，虽然miniconda大多数情况下也能用，但是conda的包管理通道很多不兼容arm架构。此外，深度学习所需的torch和torchvision的安装方式也更麻烦一点，故本文在此分享jetpack5.1.2的python环境安装教程。

安装 Archiconda 而不是 Anaconda，因为后者不提供针对 Jetson 的 wheel 文件

#下载
wget https://github.com/Archiconda/build-tools/releases/download/0.2.3/Archiconda3-0.2.3-Linux-aarch64.sh
#安装
bash Archiconda3-0.2.3-Linux-aarch64.sh -b
echo -e '\n# set environment variable for conda' >> ~/.bashrc
echo ". ~/archiconda3/etc/profile.d/conda.sh" >> ~/.bashrc
echo 'export PATH=$PATH:~/archiconda3/bin' >> ~/.bashrc
echo -e '\n# set environment variable for pip' >> ~/.bashrc
echo 'export OPENBLAS_CORETYPE=ARMV8' >> ~/.bashrc
source ~/.bashrc
#查看版本
conda --version

出现版本号说明安装成功

下载torch官方安装包

在该网站内选择jetpack5.1的python的wheel文件,这里我选的是V.14.0,因为在第一个比较顺手，其他的应该也可以。不过后续torchvision也需要安装对应torch版本的。下载需要魔法，如果下载不下来，留言我发给你们。

Jetson Zoo - eLinux.org

#安装whl
pip3 install torch-xxxxxxxxxxxxxx-linux_aarch64.whl
#检查是否安装完成
python
import torch
torch.__version__
torch.cuda.is_available()

安装torchvision，这个比较麻烦

安装各种依赖库

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

下载所需版本的torchvision源码，本文所需的为0.14.0版本，对应torch 1.14.0

https://github.com/pytorch/vision

安装torchvision

cd vision		
export BUILD_VERSION=0.14.0
python3 setup.py install --user  

一般这个时候就可以下班了，明天看吧，这个安装超级慢。

#检查是否安装完成
python
import torchvision
torchvision.__version__