Optimum documentation

Installation

You are viewing v1.18.1 version. A newer version v1.20.0 is available.
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Installation

To install 🤗 Optimum Habana, you first need to install SynapseAI and the Gaudi drivers by following the official installation guide. Then, 🤗 Optimum Habana can be installed using pip as follows:

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

To use DeepSpeed on HPUs, you also need to run the following command:

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