diff --git a/Docs/install/index.rst b/Docs/install/index.rst index 287d036ff84..f937f75e504 100644 --- a/Docs/install/index.rst +++ b/Docs/install/index.rst @@ -45,13 +45,13 @@ Use one of the following commands to install AIMET based on your choice of frame .. parsed-literal:: - python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cu121\ |whl_suffix| -f |torch_pkg_url| + python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cu121\ |torch_whl_suffix| -f |torch_pkg_url| With CPU only: .. parsed-literal:: - python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cpu\ |whl_suffix| -f |torch_pkg_url| + python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cpu\ |torch_whl_suffix| -f |torch_pkg_url| .. tab-item:: TensorFlow @@ -88,6 +88,7 @@ Use one of the following commands to install AIMET based on your choice of frame python3 -m pip install |download_url|\ |version|/aimet_onnx-|version|\+cpu\ |whl_suffix| -f |torch_pkg_url| +.. |torch_whl_suffix| replace:: \-cp310-none-any.whl .. |whl_suffix| replace:: \-cp310-cp310-manylinux_2_34_x86_64.whl .. |download_url| replace:: \https://github.com/quic/aimet/releases/download/ .. |torch_pkg_url| replace:: \https://download.pytorch.org/whl/torch_stable.html diff --git a/Docs/install/install_docker.rst b/Docs/install/install_docker.rst index a0e00efe555..a1e200fa109 100644 --- a/Docs/install/install_docker.rst +++ b/Docs/install/install_docker.rst @@ -238,7 +238,7 @@ Releases are listed at: https://github.com/quic/aimet/releases export download_url="\https://github.com/quic/aimet/releases/download/${release_tag}" # Set the wheel file name with extension, - # for example "aimet_torch-|version|\+cu121-cp310-cp310-manylinux_2_34_x86_64.whl" + # for example "aimet_torch-|version|\+cu121\ |torch_whl_suffix|" export wheel_file_name="" # NOTE: Do the following only for the PyTorch and ONNX variant packages! @@ -266,4 +266,5 @@ Environment setup source /usr/local/lib/python3.10/dist-packages/aimet_common/bin/envsetup.sh -.. _docker-install-next: + +.. |torch_whl_suffix| replace:: \-cp310-none-any.whl diff --git a/Docs/install/install_host.rst b/Docs/install/install_host.rst index 9cf664647ef..b895feaa751 100644 --- a/Docs/install/install_host.rst +++ b/Docs/install/install_host.rst @@ -116,7 +116,7 @@ For example, "|version|". Releases are listed at: https://github.com/quic/aimet/ export download_url="\https://github.com/quic/aimet/releases/download/${release_tag}" # Set the wheel file name with extension, - # for example "aimet_torch-|version|\+cu121-cp310-cp310-manylinux_2_34_x86_64.whl" + # for example "aimet_torch-|version|\+cu121\ |torch_whl_suffix|" export wheel_file_name="" # NOTE: Do the following ONLY for the PyTorch and ONNX variant packages! @@ -345,3 +345,6 @@ Pillow-SIMD is an optimized version of the Pillow Python Imaging Library. It can .. code-block:: bash source /usr/local/lib/python3.10/dist-packages/aimet_common/bin/envsetup.sh + + +.. |torch_whl_suffix| replace:: \-cp310-none-any.whl