Optimum 文件

安裝

您正在檢視的是需要從原始碼安裝。如果您想進行常規的 pip 安裝,請檢視最新的穩定版本 (v1.27.0)。
Hugging Face's logo
加入 Hugging Face 社群

並獲得增強的文件體驗

開始使用

安裝

要為 Intel® Gaudi® AI 加速器安裝 Optimum,您首先需要按照官方安裝指南安裝 Intel Gaudi 軟體和 Intel Gaudi AI 加速器驅動程式。然後,可以透過 `pip` 如下安裝適用於 Intel Gaudi 的 Optimum:

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

要將 Microsoft® DeepSpeed 與 Intel Gaudi 裝置配合使用,您還需要執行以下命令:

python -m pip install git+https://github.com/HabanaAI/DeepSpeed.git@1.21.0

為了確保您安裝了正確的 Intel Gaudi 軟體,請執行 `hl-smi` 命令以確認系統中使用的軟體版本,並在執行 DeepSpeed 安裝時應用相同的版本;請查閱 Intel Gaudi 支援矩陣並確保您使用的是適當的 DeepSpeed 版本。

< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.