安装Anaconda3

Anaconda官网：https://repo.anaconda.com/archive/

根据自己需要，复制安装包名字，以Anaconda3-2023.07-2-Linux-x86_64.sh为例

命名规则：Anaconda3-<版本号>-Linux-x86_64.sh

在终端输入命令，下载安装包

wget https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh

运行安装脚本，可以指定路径bash 安装包名字 -p PATH -u

bash Anaconda3-2023.07-2-Linux-x86_64.sh -p /software/Anaconda3 -u

安装完成后，设置环境变量，把Anaconda下的bin目录添加到环境变量中，在.bashrc文件最后添加以下代码

export PATH="/software/Anaconda3/bin:$PATH"

保存更改后，在终端输入命令

source ~/.bashrc

完成后，可输入conda -V查看版本信息，确定是否安装成功

安装PyTorch

创建虚拟环境

conda create -n pytorch python=3.9 # pytorch为虚拟环境名字，可以自己指定python版本

conda info --envs可以查看conda中安装的所有环境

激活环境

conda activate pytorch

去PyTorch官网(https://pytorch.org/)选择自己需要的版本

在终端输入nvidia-smi可以查看CUDA版本

在终端执行安装指令，弹出提示后输入y，等待完成安装即可

输入python进行python环境，输入import torch查看是否安装成功