NVIDIA JETSON AGX XAVIER 安装conda（miniforge），torch配置及torchvision配置

NVIDIA Jetson AGX Xavier 是arm架构的，无法正常无法安装Anaconda或Miniconda，但可以使用miniforge。

一、虚拟环境安装（miniforge）conda的jetson版本

miniforge的GitHub地址：https://github.com/conda-forge/miniforge

这里我建议大家可以去清华源下载Index of /github-release/conda-forge/miniforge/LatestRelease/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

下载这种格式的

 然后去文件所在目录正常安装即可

sh Miniforge-pypy3-4.8.3-4-Linux-aarch64.sh

创建环境，这里的命令和conda一样的；环境名称：ros；指定Python版本3.6（这个很重要，jetson4.6版本cuda是10.2，对应3.6的python）

conda create -n ros python=3.6

二、pytorch配置及torchvision安装

torch安装

来到官网下载PyTorch v1.7.0

PyTorch for Jetson - Jetson Nano - NVIDIA Developer Forums

到下载的目录，打开终端进行安装（注意，要在新建的虚拟环境下安装）

#先进入之前建的虚拟环境
conda activate ros

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

安装完成后，测试torch是否安装成功（先进入python指令界面）：

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.version())

 可能存在的问题：

import torch时出现“非法指令（核心已转储）”（Illegal instruction (core dumped)）

sudo vim ~/.bashrc

末尾添加，关闭并保存：

export OPENBLAS_CORETYPE=ARMV8

更新文件：

source ~/.bashrc

安装torchvision

来到github上下载0.8.1版本的torchvision：

Tags · pytorch/vision · GitHub

下载完成后解压缩，进入解压缩的目录中，执行以下命令进行安装：

python setup.py install

（实测等了很久，不要着急，只要没出错误就等就好了）

可能存在问题：不要使用sudo ，两个python版本可能不对应

测试torchvision

import torchvision
print(torchvision.__version__)

基本环境已经部署完毕，剩下的根据项目的requirement.txt来安装所需要的依赖即可