Linux服务器配置pytorch环境(自用)

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值