JETSON TX2 NX配置conda、cuda、torch、torchvision环境保姆教程

上一期已经装好了系统

本期配置一下环境

一、配置CUDA 1、ctrl+alt+T打开终端输入：

sudo gedit ~/.bashrc

2、打开完档后 在最后面添加代码

export CUDA_HOME=/usr/local/cuda-10.2 export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH export PATH=/usr/local/cuda-10.2/bin:$PATH

3、ctrl+s保存 退出文档继续在终端执行

source ~/.bashrc

4、验证是否配置cuda成功

nvcc -V

5、 如果显示如这样的图就是正常安装了， CUDA如何安装请看上文 二、conda安装

为JETSON TX2 NX的架构是aarch64,与win10,linxu不同,所以不能安装Anaconda

可以安装一个替代它的archiconda

1、下载命令

wget https://github.com/Archiconda/build-tools/releases/download/0.2.3/Archiconda3-0.2.3-Linux-aarch64.sh

2、下载好之后执行命令 bash Archiconda3-0.2.3-Linux-aarch64.sh

安装过程基本与anaconda一致，安装之后操作也一致

3、安装好之后配环境变量

sudo gedit ~/.bashrc

4、在文档最后一行添加：

export PATH=~/archiconda3/bin:$PATH

5、然后开始创建你的conda虚拟环境：

conda create -n xxx（虚拟环境名） python=3.6 #创建一个python3.6的虚拟环境 conda activate xxx #进入虚拟环境 conda deactivate #（退出虚拟环境）

6、换一下源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

7、安装pip sudo apt-get updatesudo apt-get upgradesudo apt-get dist-upgradesudo apt-get install python3-pip libopenblas-base libopenmpi-devpip3 install --upgrade pip        #如果pip已是最新，可不执行 三、开始装torch和torchvision：

安装需要去英伟达官网下载

地址:PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums

这是笔者下载好的