
pip 24.0+ 已修复对 --index-url 下 PyTorch CUDA 轮子包的缓存支持;升级 pip 后,pip install torch --index-url https://download.pytorch.org/whl/cu121 将自动命中本地缓存,避免重复下载 2.5GB 文件。
pip 24.0+ 已修复对 `--index-url` 下 pytorch cuda 轮子包的缓存支持;升级 pip 后,`pip install torch --index-url https://download.pytorch.org/whl/cu121` 将自动命中本地缓存,避免重复下载 2.5gb 文件。
在使用 pip 安装带 CUDA 支持的 PyTorch(如 torch, torchvision, torchaudio)时,官方推荐命令形如:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
该命令明确指定私有索引源(PyTorch 官方 CUDA 专用 wheel 仓库),确保安装正确版本。但在 pip --index-url 参数会绕过默认下载缓存机制——即使 pip 缓存功能已启用(默认开启),此类带 --index-url 的安装仍不会将 wheel 文件存入 pip cache,导致每次新建虚拟环境都重新下载约 2.5 GB 的二进制包。
✅ 根本解决方案:升级 pip 至 24.0 或更高版本
自 pip 24.0(2024年3月发布) 起,已正式修复 --index-url 场景下的缓存行为(见 issue #12187)。现在,只要满足以下条件,PyTorch wheel 即可被正常缓存并复用:
- pip ≥ 24.0
- 未禁用缓存(默认启用)
- 未使用 --no-cache-dir 或 PIP_NO_CACHE_DIR=1
升级命令(推荐使用 -m pip 避免路径冲突):
python -m pip install --upgrade pip # 验证版本 pip --version # 应输出 pip 24.0.x 或更高
升级后,首次安装仍将下载并缓存 wheel:
pip install torch --index-url https://download.pytorch.org/whl/cu121
随后在新虚拟环境中执行相同命令,pip 将直接从缓存读取(可通过 pip cache info 查看缓存路径,pip cache info 输出中的 Cache directory 即为位置),耗时从数分钟降至秒级。
? 验证缓存是否生效
安装后运行:
pip cache info pip cache list | grep torch
若看到类似 torch-2.3.0+cu121 的条目,说明已成功缓存。
⚠️ 注意事项
- 不要设置 PIP_DOWNLOAD_CACHE(已弃用,自 pip 20.1 起无效)或 PIP_FIND_LINKS(用于替代索引,非缓存控制);这些环境变量对当前问题无帮助。
- 若需兼容不同 CUDA 版本(如 cu118 / cu121),各版本 wheel 会独立缓存,互不干扰。
- 确保 ~/.cache/pip(Linux/macOS)或 %LOCALAPPDATA%\pip\Cache(Windows)有足够磁盘空间(建议预留 ≥5 GB)。
- 如仍遇到缓存失效,请检查是否误启用了 --no-deps、--force-reinstall 或 --no-cache-dir 等标志。
总结:无需复杂配置或第三方工具,仅需 pip install --upgrade pip 升级至 24.0+,即可让 PyTorch CUDA 包真正享受 pip 原生缓存红利,显著提升多环境开发与 CI 流程效率。











