Optimum
加入 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.