Orin NX配置

教程是在带有

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

安装完成后进行测试

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