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.23.3).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

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/[email protected]
< > Update on GitHub