解决 PyTorch 大型 wheel 下载慢、超时和反复重下的问题
在安装 PyTorch(如 torch==2.2.0+cu118)时,常遇到以下问题:
pip install中断后再次执行又从头开始下载
本文给出一个稳定、可重复、可断点的解决方案。
目录
- 问题的根本原因
- 推荐的正确安装流程(先下载,再离线安装)
1. 问题的根本原因
为什么每次都会重新下载?
pip install在安装过程中下载的文件是临时文件- 中断后,pip 不会断点续传
- 使用了
--no-cache-dir会禁用缓存 - pip 本身不适合在不稳定网络下下载 400MB+ 的 wheel
结论:不要直接用 pip install 在线下大包
2. 推荐做法(✅ 强烈推荐)
第一步:先下载所有 wheel 到本地
bash
mkdir -p wheels
pip download \
torch==2.2.0+cu118 \
torchvision==0.17.0+cu118 \
--extra-index-url https://download.pytorch.org/whl/cu118 \
-d wheels
说明:
- 这里只是 下载,不安装
- 失败可反复执行,只会补下缺的文件
第二步:从本地 wheel 离线安装
bash
pip install --no-index --find-links=./wheels \
torch==2.2.0+cu118 torchvision==0.17.0+cu118
✅ 此步骤 不会再走网络
✅ 不存在超时、重下问题
总结
一句话解决方案:
✅ 先用
pip download把 wheel 下到本地,再离线pip install,不在安装阶段下载任何文件