服务器Ubuntu系统用linux命令安装pytorch

第一步：检查一下系统GPU版本
使用命令

nvidia-smi

结果显示：

第二步：创建一个新的python虚拟环境用来安装pytorch
使用以下命令创建了一个名为mypytorch的虚拟环境

conda create -n mypytorch python=3.8

第三步：在虚拟环境中安装pytorch
（插一句，可以使用下面的命令来查看虚拟环境中的安装包）

pip list

第四步：官网的命令由于网络原因经常下载失败，所以使用清华的镜像源，直接进入虚拟环境，粘贴以下代码：

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/conda-forge/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

到这里，可以继续安装pytorch了

conda install pytorch torchvision cudatoolkit=11.0

执行完毕，安装成功
最后使用

pip list

查看以下安装包，会发现pytorch已经安装成功了

接下来可以试试import

python
import torch

结束了，撒花
退出虚拟环境,来到根目录

conda deactivate