在服务器中配置pytrch

一、链接服务器

在这里我是用的是Xshell来链接服务器，然后以下命令进入到自己的虚拟环境中

source activate 虚拟环境名称

二、查询自己环境中所拥有的包

输入以下命令来进行查询

conda list

可以看到是并没有torch的，因此我们需要安装torch

三、查询CUDA版本及对应的torch版本，并安装torch

输入以下命令来查询CUDA版本

nvidia-smi

可以看到对应的版本为12.2，在这个网址中寻找对应的torch版本，网址：Previous PyTorch Versions | PyTorch

这里没有找到对应的CUDA版本，因此选择CUDA版本为11.2所对应的torch，复制这条命令，并运行

但是，此时却报错了，这里可以看到python 版本为3.6，有可能是python版本过低，因此更新python版本为3.9.0

输入以下命令，更改python版本

conda install python==3.9.0

然后再重新运行这段命令

conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=12.1 -c pytorch -c nvidia

安装过程中，会跳转一个选择，选择y

然后等待......出现以下界面，即说明安装成功

四、检验torch是否安装成功

输入以下命令

conda list

显示已经安装成功！！！！