Linux服务器配置pytorch环境(自用)
在linux服务器上安装torch(已安装好cuda等情况下)
·
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 +文件的绝对路径 进行安装
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)