Optimum documentation

Installation

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.18.1).
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