1.查看服务器的cuda版本
nvcc --version
在服务器终端运行该代码
我的版本是11.7
【参考】:
在 Linux 和 Windows 系统下查看 CUDA 和 cuDNN 版本的方法,包括使用 nvcc 命令_cudnn版本查询-CSDN博客
2. 安装torch 【参考】在服务器中配置pytorch_服务器配置pytorch-CSDN博客
2.1 命令直接下载:从官网找自己适配的torch版本Previous PyTorch Versions | PyTorch
以我复现的代码readme为例
已知:cuda11.7 python3.9需要torch==0.4.1和torchvision==0.2.1
是比较老的版本了。。。那我就下载和另一个代码相同的版本torch=1.13 等(已经指定好镜像源)
-
pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0
发现下载的太慢了!!!镜像源也挺慢不知道为啥
-
2..2 本地轮子上传下载
-
所以转用本地轮子下载,上传到服务器https://download.pytorch.org/whl/torch_stable.html
-
-
-
接着上传到服务器
pip install +文件的绝对路径 进行安装