1、服务器中已创建自己的虚拟环境,并激活activate
2、查看服务器cuda版本
nvcc __version
3、在Previous PyTorch Versions | PyTorch找到版本对应的代码段,例如:
#CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
粘贴到服务器中运行。结果如下:
查看torch是否安装成功