Installation
============

PyPI Install
------------

The quickest way to get started with DeepSpeed-MII is to install it from `PyPI
<https://pypi.org/project/deepspeed-mii/>`_ using pip:

.. code-block:: console

   (.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:

.. code-block:: console

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

Or you can clone the repository and install:

.. code-block:: console

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