Installation

To install Optimum for Intel Gaudi, you first need to install SynapseAI and the Intel® Gaudi® drivers by following the official installation guide. Then, Optimum for Intel Gaudi 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.16.0