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 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.10.0