Skip to content

Commit

Permalink
Edit install and landing pages (#3722)
Browse files Browse the repository at this point in the history
Edited landing page, quick start page and installation pages.

Signed-off-by: Dave Welsch <[email protected]>
  • Loading branch information
dwelsch-esi authored Jan 10, 2025
1 parent 6f81d77 commit 545f33b
Show file tree
Hide file tree
Showing 5 changed files with 243 additions and 229 deletions.
27 changes: 15 additions & 12 deletions Docs/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -19,18 +19,16 @@ AIMET Documentation
API Reference <apiref/index>
Release Notes <release_notes>

The AI Model Efficiency Toolkit (AIMET) is a software toolkit designed for quantizing and compressing trained models.
The AI Model Efficiency Toolkit (AIMET) is a software toolkit for quantizing and compressing trained models. The primary objective of model optimization is to facilitate its deployment on edge devices like mobile phones or laptops.

The primary objective of model optimization is to facilitate its deployment on edge devices like mobile phones or laptops.
AIMET employs post-training and fine-tuning techniques to minimize accuracy loss during quantization and compression.

AIMET employs post-training and fine-tuning techniques to optimize trained models, aiming to minimize accuracy loss during quantization or compression.

AIMET supports models from PyTorch, TensorFlow/Keras, and ONNX frameworks.
AIMET supports models from the PyTorch, TensorFlow/Keras, and ONNX frameworks.

Quick Start
===========

To quickly install and begin using AIMET with PyTorch, see the :doc:`Quick start guide <../install/quick-start>`.
To quickly install and begin using AIMET with PyTorch, see the :doc:`Quick Start Guide <../install/quick-start>`.

Installation
============
Expand All @@ -40,32 +38,37 @@ For other install options, including for TensorFlow and ONNX platforms or to run
User Guide
==========

For a high-level explanation of how to use AIMET to optimize a model, see the :doc:`Quantization user guide <../userguide/index>`.
For a high-level explanation of how to use AIMET to optimize a model, see the :doc:`Quantization User Guide <../userguide/index>`.

Quantization Simulation Guide
=============================

For a guide to quantization simulation and its related techniques, see the doc :doc:`Quantization simulation guide <../quantsim/index>`.
Quantization simulation (QuantSim) emulates the behavior of quantized hardware in a model running on floating point hardware. For a guide to quantization simulation and its related techniques, see the :doc:`Quantization Simulation Guide <../quantsim/index>`.

Feature Guide
=============

For instructions on applying individual AIMET features, see the :doc:`Feature guide <../featureguide/index>`.
For instructions on applying individual AIMET features, see the :doc:`Feature Guide <../featureguide/index>`.

Example Notebooks
=================

To view end-to-end examples of model quantization and compression, and to download the examples in Jupyter notebook format, see :doc:`Example notebooks <../examples/index>`.
To view end-to-end examples of model quantization and compression, and to download the examples in Jupyter notebook format, see the :doc:`Example Notebooks <../examples/index>` page.

API Reference
=============

For a detailed look at the AIMET API documentation, see the :doc:`API reference <../apiref/index>`.
For a detailed look at the AIMET API, see the :doc:`API Reference <../apiref/index>`.

Release Notes
=============

For information specific to this release, see :doc:`Release notes <release_notes>`.
For information about new features in this release, see the :doc:`Release Notes <release_notes>`.

Glossary
========

See the :doc:`glossary <../glossary>` for explanations of terms and acronyms used on this website.

|
|
Expand Down
14 changes: 7 additions & 7 deletions Docs/install/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ the **.whl** files hosted at https://github.com/quic/aimet/releases.
Prerequisites
=============

The AIMET package requires the following host platform setup. Following prerequisites apply
The AIMET package requires the following host platform setup. The following prerequisites apply
to all frameworks variants.

* 64-bit Intel x86-compatible processor
Expand All @@ -21,8 +21,8 @@ to all frameworks variants.
* Nvidia driver version 455 or later (using the latest driver is recommended; both CUDA and cuDNN are supported)

.. note::
Starting with the AIMET 2 release, there is no longer a dependency on ``liblapacke``. Therefore,
you should only install the following debian package if you are still using AIMET 1.x.
Starting with the AIMET 2 release, there is no longer a dependency on ``liblapacke``.
Install the following Debian package if (and only if) you are still using AIMET 1.x.

.. code-block:: bash
Expand Down Expand Up @@ -96,16 +96,16 @@ Use one of the following commands to install AIMET based on your choice of frame
Advanced installation instructions (optional)
=============================================

Following are two ways to setup, including prerequisites and dependencies.
Following are two ways to set up, including prerequisites and dependencies.

* :ref:`On your host machine <install-host>`
* :ref:`Using our pre-built or locally built Docker images <install-docker>`
* :ref:`Using our pre-built or a locally built Docker image <install-docker>`

Installing an older version
===========================

View the release notes for older versions at https://github.com/quic/aimet/releases. Follow the
documentation corresponding to that release to select and install the appropriate AIMET package.
View the release notes for older versions at https://github.com/quic/aimet/releases. Follow instructions in the
documentation for your selected release to install the AIMET package for that release.

Building from source
====================
Expand Down
71 changes: 38 additions & 33 deletions Docs/install/install_docker.rst
Original file line number Diff line number Diff line change
Expand Up @@ -31,10 +31,10 @@ Follow the instructions below to install AIMET within a Docker container. Depend
installation choices, you will skip various sections on this page.


1. Choose your AIMET variant
----------------------------
Choose your AIMET variant
-------------------------

**1.1 Choose a variant.**
**Step 1:** Choose a variant.

Choose a variant (a combination of framework and runtime environment) from the following table.
Copy the **<variant_string>**.
Expand Down Expand Up @@ -65,7 +65,7 @@ Copy the **<variant_string>**.
- CPU
- `onnx-cpu`

**1.2 Set the AIMET_VARIANT environment variable.**
**Step 2:** Set the AIMET_VARIANT environment variable.

Set the ``AIMET_VARIANT`` shell variable to your chosen variant string.

Expand All @@ -74,21 +74,24 @@ Set the ``AIMET_VARIANT`` shell variable to your chosen variant string.
export AIMET_VARIANT=<variant_string>
2. Choose to download or build an image
---------------------------------------
Choose to download or build an image
------------------------------------

Choose one of the following options. We recommend using a prebuilt Docker image unless your
**Step 3:** Choose one of the following options. We recommend using a prebuilt Docker image unless your
installation requires custom dependencies.

- :ref:`Download a prebuilt Docker image <docker-install-download>`
- :ref:`Build a Docker image <docker-install-build>`
Continue to :ref:`Download a prebuilt Docker image (step 4)<docker-install-download>`

or

Skip to :ref:`Build a Docker image (step 5)<docker-install-build>`.

.. _docker-install-download:

3. Download a prebuilt Docker image
-----------------------------------
Download a prebuilt Docker image
--------------------------------

**3.1 Set environment variables.**
**Step 4:** Set environment variables.

Set the following shell variables to define the Docker image installation.

Expand All @@ -109,15 +112,15 @@ where:

When you start the Docker container, it will be downloaded from the image library located at **docker_image_name**.

**3.2 Skip to** :ref:`Start the Docker container <docker-install-start>`.
Skip to :ref:`Start the Docker container (step 7)<docker-install-start>`.


.. _docker-install-build:

4. Build a Docker image
-----------------------
Build a Docker image
--------------------

**4.1 Set environment variables.**
**Step 5:** Set environment variables.

Set the following shell variables to define the Docker image installation.

Expand All @@ -138,32 +141,32 @@ where:
is whatever name you want to assign the AIMET Docker container.


**4.2 Build the Docker image from code in the the AIMET repo.**
**Step 6:** Build the Docker image from code in the the AIMET repo.

.. code-block:: bash
docker build -t ${docker_image_name} -f $WORKSPACE/aimet/Jenkins/Dockerfile.${AIMET_VARIANT} .
.. _docker-install-start:

5. Start the docker container
-----------------------------
Start the docker container
--------------------------

**5.1 Check that a Docker container named $docker_container_name is not already running. Remove the container if it is.**
**Step 7:** Check that a Docker container named $docker_container_name is not already running. Remove the container if it is.

.. code-block:: bash
docker ps -a | grep ${docker_container_name} && docker kill ${docker_container_name}
**5.2 Specify a port to use for port forwarding if you plan to run the Visualization APIs (optional).**
**Step 8:** (optional) Specify a port to use for port forwarding if you plan to run the Visualization APIs.

.. code-block:: bash
port_id="<port-number>"
where **<port-number>** is any unused port on the host.

**5.3 Run the Docker container.**
**Step 9:** Run the Docker container.

.. code-block:: bash
Expand Down Expand Up @@ -202,35 +205,37 @@ As a convenience, the following block contains the *first line* of the Docker ru
nvidia-docker run --rm -it -u $(id -u ${USER}):$(id -g ${USER}) \
# CPU only, without port forwarding:
docker run --rm -it -u $(id -u ${USER}):$(id -g ${USER})
docker run --rm -it -u $(id -u ${USER}):$(id -g ${USER}) \
6. Install AIMET packages
-------------------------
Install AIMET packages
----------------------

**Choose an option to install the AIMET package on the Docker container.**

1. From PyPI (PyTorch only)
2. Any framework variant (hosted **.whl** files)

**6.1 To install the most recent PyTorch AIMET package with GPU support (the most common option) from PyPI, type the following commands in the Docker container.**
**Step 10:** To install the most recent PyTorch AIMET package with GPU support (the most common option) from PyPI, type the following commands in the Docker container.

.. code-block:: bash
python3 -m pip install aimet-torch
**6.2 To install the latest version of any AIMET variant from the.whl files.**
Skip to :ref:`Environment setup (step 12) <docker-install-setup>`.

**Step 11:** To install the latest version of any AIMET variant from the .whl files, follow the substeps below.

6.2.1 Select the release tag for the version you want to install, for example, "|version|".
**Step 11.1:** Select the release tag for the version you want to install, for example, "|version|".

Releases are listed at: https://github.com/quic/aimet/releases

- Identify the .whl file corresponding to the package variant that you want to install
- Continue with the instructions below to install AIMET from the .whl file

6.2.2 Set the package details.
**Step 11.2:** Set the package details.

.. parsed-literal::
# Set the release tag, for example "|version|"
export release_tag="<version release tag>"
Expand All @@ -244,7 +249,7 @@ Releases are listed at: https://github.com/quic/aimet/releases
# NOTE: Do the following only for the PyTorch and ONNX variant packages!
export find_pkg_url_str="-f https://download.pytorch.org/whl/torch_stable.html"
6.2.3 Install the selected AIMET package.
**Step 11.3:** Install the selected AIMET package.

.. note::

Expand All @@ -258,9 +263,9 @@ Releases are listed at: https://github.com/quic/aimet/releases
.. _docker-install-setup:

Environment setup
=================
-----------------

**Run the environment setup script to set common environment variables.**
**Step 12:** Run the environment setup script to set common environment variables.

.. code-block:: bash
Expand Down
Loading

0 comments on commit 545f33b

Please sign in to comment.