安装PyTorch 2.4.1+cu121（本机电脑cuda支持12.7）

程序改变世界&

4023人浏览 · 2025-01-13 12:25:58

程序改变世界& · 2025-01-13 12:25:58 发布

在这里插入图片描述
您可以尝试安装 CUDA 12.1 驱动程序，或者使用与 CUDA 12.1 兼容的 PyTorch 2.4.1。请使用以下命令来安装 PyTorch 2.4.1+cu121：

pip install torch==2.4.1+cu121 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

在这里插入图片描述

FlagOS智算系统软件栈

欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

大模型新拐点：FlagOS+Engram 开启算存协同新时代

本文首先介绍 Engram 的核心思想，然后阐述基于 FlagOS 系统软件栈中的训练插件完成的 Engram 架构全链路复现。在此基础上，重点展示 FlagOS 针对 Engram 进行的三大工程优化。实验量化结果表明，FlagOS 对 Engram 的优化在保持额外负载为零的前提下，使吞吐最高提升近 150%。