一、链接服务器
在这里我是用的是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
显示已经安装成功!!!!