Installation

PyPI Install

The quickest way to get started with DeepSpeed-MII is to install it from PyPI using pip:

(.venv) $ pip install deepspeed-mii

Source Install

If you want the latest changes on the main repository branch, you can use pip to install from source:

(.venv) $ pip install git+https://github.com/deepspeedai/DeepSpeed-MII.git

Or you can clone the repository and install:

(.venv) $ git clone https://github.com/deepspeedai/DeepSpeed-MII.git
(.venv) $ pip install ./DeepSpeed-MII