目录
1.下载Anconda
不同版本在这里查看,下载你需要的版本
wget https://repo.anaconda.com/archive/Anaconda3-2023.09-0-Linux-x86_64.sh
2.先赋权再安装
chmod +x Anaconda3-2023.09-0-Linux-x86_64.sh
./Anaconda3-2023.09-0-Linux-x86_64.sh
按回车继续,出现一堆文字,疯狂按回车(或按空格更快),直到出现"Do you accept the license terms?",输入yes继续
询问你安装路径,没问题就按回车继续,直到出现"Thank you for installing Anaconda3!"
输入anaconda -V查看是否安装成功,如果输出版本号则表示成功,直接进行第3步。如果报conda: command not found问题,输入下面两行:
echo 'export PATH="~/anaconda3/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc
3.创建虚拟环境
conda create -n torch python=3.8
*(torch是我自定义的环境名,你也可以改成其他名称)期间询问y/n,直接y即可
4.激活环境
conda activate torch
退出当前环境为 conda deactivate
5.安装PyTorch
去PyTorch官网 查询所需对应命令,我这里安装的是GPU版本(注意Package选项需要选择Pip而不是Conda,Conda对应命令安装的实则是CPU版本)
*这里需要注意,Compute Platform对应CUDA版本号应小于显卡CUDA版本号
显卡CUDA版本可以调用watch nvidia-smi指令查询,如红框所示:
在虚拟环境下输入对应的PyTorch安装指令,进行安装(需要一段时间)
安装完成后,依次输入:
python
import torch
torch.cuda.is_available()
返回True则大功告成!