PyTorch国内镜像源下载速度慢怎么办？

Question

在使用 pip 安装 PyTorch 时，即使配置了国内镜像源（如清华、阿里云），下载速度依然缓慢，主要原因在于 PyTorch 官方包托管在 `download.pytorch.org` 而非 PyPI，因此 pip 的镜像源对 `.whl` 文件无效。用户常误以为更换 pip 源即可加速，但实际上这些镜像不代理 PyTorch 的专属存储地址。解决该问题需直接配置 torch 相关索引 URL，或使用 conda 配合国内镜像源（如清华 TUNA），以显著提升下载速度。

Answer 1

1. 问题背景与现象描述

在使用 pip install torch 安装 PyTorch 时，许多开发者发现即使已配置了国内镜像源（如清华 TUNA、阿里云、豆瓣等），下载速度依然极其缓慢。典型表现是：

• 下载进度条长时间停滞
• 平均速度低于 100KB/s
• 频繁出现超时或连接中断
• download.pytorch.org 域名持续出现在日志中

该现象普遍存在于中国大陆地区的开发环境中，尤其在企业级部署和 CI/CD 流水线中尤为突出。

2. 根本原因分析

PyTorch 的官方二进制包（.whl 文件）并非托管在标准的 Python Package Index (PyPI) 上，而是由其专属 CDN 托管于 download.pytorch.org。这意味着：

源类型	是否代理 download.pytorch.org	代表源
PyPI 官方源	是	pypi.org
国内 pip 镜像	否	mirrors.tuna.tsinghua.edu.cn/pypi
Conda 国内镜像	是（部分）	mirrors.tuna.tsinghua.edu.cn/anaconda

因此，仅修改 pip 的 index-url 并不能加速 PyTorch 包的下载，因为这些镜像不反向代理非 PyPI 路径。

3. 解决方案路径对比

目前主流解决方案分为两大类：基于 pip 的索引 URL 配置 和 基于 conda 的镜像加速。以下是详细对比：

# 方法一：直接指定 torch 索引 URL
pip install torch torchvision torchaudio \
    -f https://download.pytorch.org/whl/cu118

# 方法二：使用清华镜像加速 torch 下载
pip install torch torchvision torchaudio \
    -f https://pypi.tuna.tsinghua.edu.cn/simple/torch/

注意：第二种方式需确保镜像站已同步 torch 资源，目前清华 TUNA 提供了完整的 torch 镜像服务。

4. 推荐实践流程图

graph TD
    A[开始安装 PyTorch] --> B{使用 pip 还是 conda?}
    B -->|pip| C[检查是否配置 torch 特定索引]
    B -->|conda| D[配置清华 conda 源]
    C --> E[添加 -f https://pypi.tuna.tsinghua.edu.cn/simple/torch/]
    D --> F[运行 conda install pytorch cudatoolkit=11.8 -c pytorch]
    E --> G[执行安装命令]
    F --> G
    G --> H[验证安装结果]

5. conda 镜像配置实操

对于长期使用深度学习框架的团队，推荐使用 conda 并配置国内镜像源。步骤如下：

1. 编辑 ~/.condarc 文件
2. 添加清华 TUNA 镜像通道
3. 设置默认 channel_alias
4. 启用 show_channel_urls
5. 清除缓存并测试速度

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda

6. 高级优化建议

针对企业级环境，可进一步采取以下措施：

• 搭建本地 pip 与 conda 私有镜像缓存服务器
• 使用 Nexus 或 Artifactory 对外网资源进行代理
• 预构建包含 PyTorch 的 Docker 镜像并推送到内网 registry
• 通过 CDN 加速 download.pytorch.org 的访问（需合规性评估）
• 监控网络链路质量，识别 DNS 污染或路由绕行问题

这些策略可显著提升大规模部署效率，降低对外部网络的依赖。