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 +文件的绝对路径 进行安装

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐