教程是在带有

1. 首先换源，详见Ubuntu20.04更换清华源/阿里源，以及aarch64报错处理-CSDN博客

2. 安装jetpack组件

sudo apt install nvidia-jetpack

3.配置conda源

nvidia的一些命令，在orgin nx里面用不了，比如nvidia-smi，nvcc等

首先安装anaconda，要求使用清华源的，aarch64和普通的源配置不太一样

首先编辑

sudo nano ~/.condarc

将内容替换为以下内容

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

4. 安装pytroch

查看网页为 Jetson 平台安装 PyTorch - NVIDIA 文档

sudo apt-get -y update

sudo apt-get install -y  python3-pip libopenblas-dev

然后创建一个环境

conda create -n torch python=3.8

激活环境，进行以下步骤

export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v511/pytorch/torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl

python3 -m pip install --no-cache $TORCH_INSTALL

安装完成后进行测试

如果是以下输出则证明大功告成，可以使用！