1、下载安装适配aarch64架构的miniforge或者miniconda(Anaconda轻量化版本)
方法一:使用终端
运行
wget https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-Linux-aarch64.sh
or
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh
方法二:去网址下在miniforge.sh/miniconda.sh文件
miniforge下载网址:https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-Linux-aarch64.sh
miniconda下载网址:https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh
清华源下载:https://mirrors.tuna.tsinghua.edu.cn/github-release/conda-forge/miniforge/
2、安装miniforge/miniconda
1)安装
方法一情况下直接运行下面程序
bash Miniforge3-Linux-aarch64.sh
方法二情况下
若在download目录下,打开下载地址
(1)可直接打开终端,cd 进入:
cd /home/用户名/download/
(2)或在download文件夹内打开终端
然后运行
bash Miniforge3-Linux-aarch64.sh
Miniforge3-Linux-aarch64.sh为下载好的.sh文件
2)验证
按照提示完成安装后,重启终端。输入
conda
#or
conda --version
#or
echo $PATH
若出现 conda:command not found
或 在路径中未找到: /home/用户名/miniconda3/bin/
则进行第三步,将conda指令添加到路径中去
注:若正常运行,则无需进行第三步
3)添加路径
首先,进入conda的文件位置
cd ~/miniconda3/bin/
#or
cd ~/miniforge3/bin/
然后,使用 pwd 获取路径,得到
/home/用户名/miniforge3/bin/
其次,使用nano编译器编译bashrc文件,输入下面指令后进入最下面,
nano ~/.bashrc
#进入底端,
在#<<<conda initialize<<< 下面添加路径
export PATH="/home/用户名/miniforge3/bin/:$PATH"
最后使用Ctrl+X之后按Y;选择yes保存操作;按enter键保存在默认位置。
至此,bashrc文件修改成功。
输入source ~/.bashrc
,即可正常使用conda指令
miniconda安装成功
3. 配置 PyTorch 环境
1)查看CUDA版本
方法 1:通过命令行
在终端或命令提示符中运行以下命令:
nvcc --version
输出中会显示 CUDA 的版本