如何让 PyTorch(CUDA 版)被 pip 正确缓存以加速多环境安装

小强酱_3463

小强酱_3463

2026-07-14

421人浏览

原创

如何让 PyTorch(CUDA 版)被 pip 正确缓存以加速多环境安装

pip 24.0+ 默认启用并正确支持带 --index-url 的 pytorch cuda 轮子(wheel)缓存,升级 pip 即可解决重复下载问题,无需额外配置。

pip 24.0+ 默认启用并正确支持带 --index-url 的 pytorch cuda 轮子(wheel)缓存,升级 pip 即可解决重复下载问题,无需额外配置。

在多虚拟环境开发中频繁安装 torch + CUDA 支持(如 --index-url https://download.pytorch.org/whl/cu121)时,你可能遇到:每次新建 venv 都重新下载约 2.5 GB 的 wheel 文件,即使 pip 官方文档明确说明其默认启用缓存(pip caching docs),但 torch 却始终未出现在 pip cache info 或 pip cache list 中。

根本原因在于 pip 版本兼容性:
在 pip ≤ 23.x 中,当使用 --index-url(尤其是非 PyPI 默认源的显式索引地址)时,部分版本存在缓存逻辑缺陷——pip 未能将远程 wheel 的哈希与缓存键正确关联,导致下载后不写入缓存,或后续安装时无法命中缓存。该问题已在 pip 24.0 及更高版本中修复(参见 pip#12396 等相关 issue)。

✅ 解决方案:升级 pip 至 24.0+

python -m pip install --upgrade pip

升级完成后,验证版本:

pip --version  # 应输出类似:pip 24.0.1 from ...

随后执行标准安装命令(无需修改):

百度文档解析pipeline-parser
百度文档解析pipeline-parser

调用百度文档解析API,支持18+格式,提取文本、表格、版面分析、OCR识别及RAG文档分块。适用于文档解析、文本/表格提取、结构分析、扫描件处理。触发词:文档解析、PDF解析、Word解析、表格提取、OCR、文档分析、提取文本、文档结构、扫描识别。

下载
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

✅ 第一次安装会正常下载并自动缓存至本地(路径可通过 pip cache dir 查看,通常为 ~/.cache/pip);
✅ 后续在其他 venv 中执行相同命令,pip 将直接从缓存复制 wheel,跳过网络下载,耗时从分钟级降至秒级。

? 验证缓存是否生效:

# 查看缓存目录及大小
pip cache info

# 列出已缓存的包(应包含 torch-*.whl)
pip cache list | grep torch

# 清理(可选,调试时使用)
# pip cache purge

⚠️ 注意事项:

  • 不要设置 PIP_DOWNLOAD_CACHE(已弃用)或 PIP_FIND_LINKS(干扰依赖解析),现代 pip 无需这些变量;
  • --index-url 本身不会禁用缓存,问题仅存在于旧版 pip 的实现缺陷;
  • 若使用 --no-cache-dir 或 CI 环境中显式禁用了缓存,请确保移除该标志;
  • 多 CUDA 版本(如 cu118/cu121/cu124)对应不同 wheel,各自独立缓存,互不冲突。

升级 pip 是最轻量、最可靠且符合官方推荐的解法。从此,无论创建多少个 Python 3.10 venv,PyTorch CUDA 版本都只需下载一次,大幅提升本地开发与 CI 构建效率。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

pytorch pip

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
pip安装使用方法
pip安装使用方法

安装步骤:1、确保Python已经正确安装在您的计算机上;2、下载“get-pip.py”脚本;3、按下Win + R键,然后输入cmd并按下Enter键来打开命令行窗口;4、在命令行窗口中,使用cd命令切换到“get-pip.py”所在的目录;5、执行安装命令;6、验证安装结果即可。大家可以访问本专题下的文章,了解pip安装使用方法的更多内容。

2023.10.09

2610

5

更新pip版本
更新pip版本

更新pip版本方法有使用pip自身更新、使用操作系统自带的包管理工具、使用python包管理工具、手动安装最新版本。想了解更多相关的内容,请阅读专题下面的文章。

2024.12.20

6043

6

pip设置清华源
pip设置清华源

设置方法:1、打开终端或命令提示符窗口;2、运行“touch ~/.pip/pip.conf”命令创建一个名为pip的配置文件;3、打开pip.conf文件,然后添加“[global];index-url = https://pypi.tuna.tsinghua.edu.cn/simple”内容,这将把pip的镜像源设置为清华大学的镜像源;4、保存并关闭文件即可。

2024.12.23

7602

11

python升级pip
python升级pip

本专题整合了python升级pip相关教程,阅读下面的文章了解更多详细内容。

2025.07.23

3727

16

pytorch是干嘛的
pytorch是干嘛的

pytorch是一个基于python的深度学习框架,提供以下主要功能:动态图计算,提供灵活性。强大的张量操作,实现高效处理。自动微分,简化梯度计算。预构建的神经网络模块,简化模型构建。各种优化器,用于性能优化。想了解更多pytorch的相关内容,可以阅读本专题下面的文章。

2024.05.29

1963

8

Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习

PyTorch 是一种用于构建深度学习模型的功能完备框架,是一种通常用于图像识别和语言处理等应用程序的机器学习。 使用Python 编写,因此对于大多数机器学习开发者而言,学习和使用起来相对简单。 PyTorch 的独特之处在于,它完全支持GPU,并且使用反向模式自动微分技术,因此可以动态修改计算图形。

2025.12.22

68

5

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

2026.09.30

20

10

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

2026.09.30

40

14

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

2026.09.30

20

12

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程