PyTorch 加速下载,使用国内镜像源
PyTorch 是一个非常流行的深度学习框架,但是,官方的 PyTorch 下载地址在国外,访问速度很慢。
由于众所周知的原因,下载速度和链接很不稳定。网上给的答案都是去使用阿里云镜像,然而阿里云的镜像好久没更新了,根本没法满足现在的需求。
所以真的没办法了吗?
使用南京大学镜像源
这里介绍一个比较冷门但质量稳定的镜像源:南京大学开源镜像站。它们一直在同步官方最新的 PyTorch 镜像源,并且速度非常快。
使用方法就是将安装命令中的 https://download.pytorch.org/whl/
替换为 https://mirrors.nju.edu.cn/pytorch/whl/
。
比如,你要安装 CUDA 12.6 的 PyTorch,官方的安装命令是:
1 | pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126 |
你只需要改成:
1 | pip3 install torch torchvision torchaudio --index-url https://mirrors.nju.edu.cn/pytorch/whl/cu126 |
然后就可以了。
目前的问题
截止 2025年 5 月 22 日,该镜像站仍然存在一些问题:
- CUDA 12.8 (
cu128
)版本的 PyTorch 无法安装。 - 最后一次更新是一个月前,而且最近几次的同步都失败了,不知道为什么。据说是因为 PyTorch 上游调整了镜像站的访问。
但是这也是国内唯一一个比较新的 PyTorch 镜像站了。有总比没有好。
速度
速度还是非常快的,稳定性和其他镜像站差不多。
我记得去年我还和南大的朋友聊起他们的镜像站,他说他们南大学生自己都不用,因为机房的网络设备都是十几年前的老古董了……当时南大镜像站的下载速度只有 4MB/s。
- 标题: PyTorch 加速下载,使用国内镜像源
- 作者: GT610
- 创建于 : 2025-05-22 22:33:03
- 更新于 : 2025-05-25 17:19:00
- 链接: https://gt-610.github.io/2025/05/22/pytorch-mirrors/
- 版权声明: 本文章采用 CC BY 4.0 进行许可。
评论