step1 : 查看服务器中显卡版本以及对应的CUDA版本
在XShell中输入指令查看服务器信息 :nvidia-smi(下载的cuda版本数小于你自身服务器中cuda版本即可)
step2 :去官网找到对应的pytorch版本的下载指令(要与python和CUDA版本对应下载,不然会无法运行),本人下载的是与python3.6对应的版本。官网网址如下:
不同版本的torch与python版本的兼容性,下表来自官网文档,网址如下:项目概览 - vision - GitCodehttps://gitcode.com/pytorch/vision/overview?utm_source=csdn_github_accelerator&isLogin=1
step3 : 先进入服务器中你对应的环境中,然后将step2中的下载指令输入到命令行中,进行下载
step4 :下载完成后,验证GPU版本的torch是否安装成功。依次输入的指令如下
- python
- import torch
- torch.cuda.is_available()
step5 :输入pip list 指令查看已下载的数据包
补充:使用xftp将本地已有数据包直接导入到服务器中。