wget https://repo.anaconda.com/archive/Anaconda3-2023.09-0-Linux-x86_64.sh
wget https://mirrors.aliyun.com/anaconda/archive/Anaconda3-2023.09-0-Linux-x86_64.sh
bash Anaconda3-2023.09-0-Linux-x86_64.sh -p /home/ubuntu_18/anaconda3 -u
打开/home/username/.bashrc文件,添加环境变量(username是你自己的用户名)
export PATH="/home/username/anaconda3/bin:$PATH"
conda config --set auto_activate_base true
source ~/.bashrc
conda --version
nvidia-smi
nvcc -V
sudo apt install nvidia-cuda-toolkit
再次执行nvcc -V:得到如下结果
conda create -n pytorch1.2cuda11.3 python=3.8
激活新创建的环境(新版本再也不能conda activate了,使用source activate替代)
source activate pytorch1.2cuda11.3
关闭新创建的环境
conda deactivate
查看虚拟环境列表
conda env list
删除虚拟环境
conda env remove -p /home/ubuntu_18/anaconda3/envs/pytorch1.2cuda11.3
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
import torch
torch.cuda.is_available()
最终安装完成!!!