讲师中心 微信公众号
AI工具推荐 视频效率加速

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

小强酱_3463

小强酱_3463

发布时间:2026-07-14 20:59:25

|

421人浏览过

|

来源于php中文网

原创

如何让 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 ...

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

PipiAds Dropshipping Product Research
PipiAds Dropshipping Product Research

通过 TikTok、Facebook 广告信号、产品详情和 PipiAds 店铺情报研究代发货产品和店铺。

下载
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 构建效率。

热门AI工具

更多
DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

相关专题

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

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

2610

2023.10.09

更新pip版本
更新pip版本

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

6003

2024.12.20

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

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

7562

2024.12.23

python升级pip
python升级pip

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

3707

2025.07.23

pytorch是干嘛的
pytorch是干嘛的

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

1963

2024.05.29

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

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

68

2025.12.22

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

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

20

2026.09.30

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

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

0

2026.09.30

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

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

0

2026.09.30

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn