1.查看CUDA的version
nvidia-smi
2.在下面这个官网地址中根据CUDA版本和python版本找到对应的命令
https://pytorch.org/get-started/previous-versions/
3.
# CUDA 11.1
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
打开最后的链接,然后根据文件名字找到对应的三个文件分别是(!这里要注意需要下载linux系统对应的文件)
torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl,torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl,torchvision-0.9.1+cu111-cp38-cp38-linux_x86_64.whl
右键复制链接,到迅雷中下载
4.下载完之后上传到九天服务器的控制台界面
直接放在根目录下,路径比较简单
# 进入虚拟环境中
conda activate 虚拟环境名字
依次执行安装命令(注意顺序一致)
pip install /root/torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl
pip install /root/torchvision-0.9.1+cu111-cp38-cp38-linux_x86_64.whl
pip install /root/torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl
5.安装成功