Optimum 文档

安装

您正在查看 主分支 版本,需要从源代码安装。如果您希望使用常规的 pip 安装,请查看最新的稳定版本 (v1.23.1).
Hugging Face's logo
加入 Hugging Face 社区

并获得增强型文档体验

开始使用

安装

要为 Intel Gaudi 安装 Optimum,您首先需要按照官方 安装指南 安装 SynapseAI 和 Intel® Gaudi® 驱动程序。然后,可以使用以下方法使用 pip 安装适用于 Intel Gaudi 的 Optimum

python -m pip install --upgrade-strategy eager optimum[habana]

要在 HPU 上使用 DeepSpeed,您还需要运行以下命令

python -m pip install git+https://github.com/HabanaAI/[email protected]
< > 更新 在 GitHub 上