diff --git a/docs/install/3rd-party/pytorch-install.rst b/docs/install/3rd-party/pytorch-install.rst index 6ba56000..c02ea3a9 100644 --- a/docs/install/3rd-party/pytorch-install.rst +++ b/docs/install/3rd-party/pytorch-install.rst @@ -80,10 +80,6 @@ should perform as tested and mitigate potential installation issues. See This will automatically download the image if it does not exist on the host. You can also pass the ``-v`` argument to mount any data directories from the host onto the container. -.. |hr| raw:: html - -
- .. _pytorch-docker-support: Docker image support @@ -95,198 +91,240 @@ validated for ROCm 6.2. .. tab-set:: - .. tab-item:: Ubuntu 22.04 - - Tag - `rocm/pytorch:rocm6.2.1_ubuntu22.04_py3.10_pytorch_release_2.1.2 `_ - - Inventory - * `ROCm 6.2.1 `_ - * `Python 3.10 `_ - * `PyTorch 2.1.2 `_ - * `Apex 1.1.0 `_ - * `torchvision 0.16.1 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.14.1 `_ - * `OMPI 4.1.5 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2_ubuntu22.04_py3.10_pytorch_release_2.3.0 `_ - - Inventory - * `ROCm 6.2 `_ - * `Python 3.10 `_ - * `PyTorch 2.3.0 `_ - * `Apex 1.3.0 `_ - * `torchvision 0.18.0 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.14.1 `_ - * `OMPI 4.1.5 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_release_2.2.1 `_ - - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `PyTorch 2.2.1 `_ - * `Apex 1.2.0 `_ - * `torchvision 0.17.1 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.14.1 `_ - * `OMPI 4.1.5 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_1.13.1 `_ - - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `PyTorch 1.13.1 `_ - * `Apex 1.0.0 `_ - * `torchvision 0.14.0 `_ - * `TensorBoard 2.17.0 `_ - * `MAGMA `_ - * `UCX 1.14.1 `_ - * `OMPI 4.1.5 `_ - * `OFED 5.4.3 `_ - - .. tab-item:: Ubuntu 20.04 - - Tag - `rocm/pytorch:rocm6.2.1_ubuntu20.04_py3.9_pytorch_release_2.3.0 `_ (`latest `_) - - .. note:: - - As of ROCm 6.2.1, ``rocm/pytorch:latest`` points to a Docker image with the latest ROCm tested - release version of PyTorch (for example, version 2.3), similar to ``rocm/pytorch:latest-release`` tag. See - :ref:`Using a Docker image with PyTorch pre-installed ` for more information. - - Inventory - * `ROCm 6.2.1 `_ - * `Python 3.9 `_ - * `PyTorch 2.3.0 `_ - * `Apex 1.3.0 `_ - * `torchvision 0.18.0 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.10.1 `_ - * `OMPI 4.0.3 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2.1_ubuntu20.04_py3.9_pytorch_release_2.2.1 `_ - - Inventory - * `ROCm 6.2.1 `_ - * `Python 3.9 `_ - * `PyTorch 2.2.1 `_ - * `Apex 1.2.0 `_ - * `torchvision 0.17.1 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.10.1 `_ - * `OMPI 4.0.3 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2.1_ubuntu20.04_py3.9_pytorch_release_1.13.1 `_ - - Inventory - * `ROCm 6.2.1 `_ - * `Python 3.9 `_ - * `PyTorch 1.13.1 `_ - * `Apex 1.0.0 `_ - * `torchvision 0.14.0 `_ - * `TensorBoard 2.17.0 `_ - * `MAGMA `_ - * `UCX 1.10.0 `_ - * `OMPI 4.0.3 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.3.0 `_ - - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `PyTorch 2.3.0 `_ - * `Apex 1.3.0 `_ - * `torchvision 0.18.0 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.10.1 `_ - * `OMPI 4.0.3 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_2.2.1 `_ - - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `PyTorch 2.2.1 `_ - * `Apex 1.2.0 `_ - * `torchvision 0.17.1 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.10.0 `_ - * `OMPI 4.0.3 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.1.2 `_ - - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `PyTorch 2.1.2 `_ - * `Apex 1.1.0 `_ - * `torchvision 0.16.1 `_ - * `TensorBoard 2.13.0 `_ - * `MAGMA `_ - * `UCX 1.10.0 `_ - * `OMPI 4.0.3 `_ - * `OFED 5.4.3 `_ - - |hr| - - Tag - `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_1.13.1 `_ - - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `PyTorch 1.13.1 `_ - * `Apex 1.0.0 `_ - * `torchvision 0.14.0 `_ - * `TensorBoard 2.17.0 `_ - * `MAGMA `_ - * `UCX 1.10.0 `_ - * `OMPI 4.0.3 `_ - * `OFED 5.4.3 `_ + .. tab-item:: ROCm 6.2.1 + + .. tab-set:: + + .. tab-item:: PyTorch 2.3.0 + + .. tab-set:: + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/pytorch:rocm6.2.1_ubuntu20.04_py3.9_pytorch_release_2.3.0 `_ (`latest `_) + + .. note:: + + As of ROCm 6.2.1, ``rocm/pytorch:latest`` points to a Docker image with the latest ROCm tested + release version of PyTorch (for example, version 2.3), similar to ``rocm/pytorch:latest-release`` tag. See + :ref:`Using a Docker image with PyTorch pre-installed ` for more information. + + Inventory + * `ROCm 6.2.1 `_ + * `Python 3.9 `_ + * `PyTorch 2.3.0 `_ + * `Apex 1.3.0 `_ + * `torchvision 0.18.0 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.10.1 `_ + * `OMPI 4.0.3 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: PyTorch 2.2.1 + + .. tab-set:: + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/pytorch:rocm6.2.1_ubuntu20.04_py3.9_pytorch_release_2.2.1 `_ + + Inventory + * `ROCm 6.2.1 `_ + * `Python 3.9 `_ + * `PyTorch 2.2.1 `_ + * `Apex 1.2.0 `_ + * `torchvision 0.17.1 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.10.1 `_ + * `OMPI 4.0.3 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: PyTorch 2.1.2 + + .. tab-set:: + + .. tab-item:: Ubuntu 22.04 + + Tag + `rocm/pytorch:rocm6.2.1_ubuntu22.04_py3.10_pytorch_release_2.1.2 `_ + + Inventory + * `ROCm 6.2.1 `_ + * `Python 3.10 `_ + * `PyTorch 2.1.2 `_ + * `Apex 1.1.0 `_ + * `torchvision 0.16.1 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.14.1 `_ + * `OMPI 4.1.5 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: PyTorch 1.13.1 + + .. tab-set:: + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/pytorch:rocm6.2.1_ubuntu20.04_py3.9_pytorch_release_1.13.1 `_ + + Inventory + * `ROCm 6.2.1 `_ + * `Python 3.9 `_ + * `PyTorch 1.13.1 `_ + * `Apex 1.0.0 `_ + * `torchvision 0.14.0 `_ + * `TensorBoard 2.17.0 `_ + * `MAGMA `_ + * `UCX 1.10.0 `_ + * `OMPI 4.0.3 `_ + * `OFED 5.4.3 `_ + + + .. tab-item:: ROCm 6.2.0 + + .. tab-set:: + + .. tab-item:: PyTorch 2.3.0 + + .. tab-set:: + + .. tab-item:: Ubuntu 22.04 + + Tag + `rocm/pytorch:rocm6.2_ubuntu22.04_py3.10_pytorch_release_2.3.0 `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.10 `_ + * `PyTorch 2.3.0 `_ + * `Apex 1.3.0 `_ + * `torchvision 0.18.0 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.14.1 `_ + * `OMPI 4.1.5 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.3.0 `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `PyTorch 2.3.0 `_ + * `Apex 1.3.0 `_ + * `torchvision 0.18.0 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.10.1 `_ + * `OMPI 4.0.3 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: PyTorch 2.2.1 + + .. tab-set:: + + .. tab-item:: Ubuntu 22.04 + + Tag + `rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_release_2.2.1 `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `PyTorch 2.2.1 `_ + * `Apex 1.2.0 `_ + * `torchvision 0.17.1 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.14.1 `_ + * `OMPI 4.1.5 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_2.2.1 `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `PyTorch 2.2.1 `_ + * `Apex 1.2.0 `_ + * `torchvision 0.17.1 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.10.0 `_ + * `OMPI 4.0.3 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: PyTorch 2.1.2 + + .. tab-set:: + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.1.2 `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `PyTorch 2.1.2 `_ + * `Apex 1.1.0 `_ + * `torchvision 0.16.1 `_ + * `TensorBoard 2.13.0 `_ + * `MAGMA `_ + * `UCX 1.10.0 `_ + * `OMPI 4.0.3 `_ + * `OFED 5.4.3 `_ + + .. tab-item:: PyTorch 1.13.1 + + .. tab-set:: + + .. tab-item:: Ubuntu 22.04 + + Tag + `rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_1.13.1 `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `PyTorch 1.13.1 `_ + * `Apex 1.0.0 `_ + * `torchvision 0.14.0 `_ + * `TensorBoard 2.17.0 `_ + * `MAGMA `_ + * `UCX 1.14.1 `_ + * `OMPI 4.1.5 `_ + * `OFED 5.4.3 `_ + + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_1.13.1 `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `PyTorch 1.13.1 `_ + * `Apex 1.0.0 `_ + * `torchvision 0.14.0 `_ + * `TensorBoard 2.17.0 `_ + * `MAGMA `_ + * `UCX 1.10.0 `_ + * `OMPI 4.0.3 `_ + * `OFED 5.4.3 `_ .. _install_pytorch_wheels: .. _using-wheels-package: @@ -367,12 +405,12 @@ wheels command, you must select **Linux**, **Python**, **pip**, and **ROCm** in .. note:: - The following command uses the ROCm 6.3 PyTorch wheel. If you want a different version of ROCm, + The following command uses the ROCm 6.2 PyTorch wheel. If you want a different version of ROCm, modify the command accordingly. .. code-block:: bash - pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3/ + pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2/ 4. (Optional) Use MIOpen kdb files with ROCm PyTorch wheels. @@ -396,11 +434,11 @@ wheels command, you must select **Linux**, **Python**, **pip**, and **ROCm** in .. code-block:: bash - #Optional: replace 'gfx90a' with your architecture and 6.3 with your preferred ROCm version + #Optional: replace 'gfx90a' with your architecture and 6.2 with your preferred ROCm version export GFX_ARCH=gfx90a #Optional - export ROCM_VERSION=6.3 + export ROCM_VERSION=6.2 ./install_kdb_files_for_pytorch_wheels.sh @@ -523,7 +561,7 @@ maintainers and installs all the required dependencies, including: Where: * ```` = ``ubuntu20.04`` (or ``focal``), ``ubuntu22.04`` (or ``jammy``) - * ```` = ``6.0``, ``6.1``, ``6.2``, ``6.3`` + * ```` = ``6.0``, ``6.1``, ``6.2`` * ```` = ``3.8`` - ``3.11`` To verify that your image was successfully created, run: diff --git a/docs/install/3rd-party/tensorflow-install.rst b/docs/install/3rd-party/tensorflow-install.rst index d2276d7c..6a7af539 100644 --- a/docs/install/3rd-party/tensorflow-install.rst +++ b/docs/install/3rd-party/tensorflow-install.rst @@ -90,104 +90,136 @@ validated for ROCm 6.2. .. tab-set:: - .. tab-item:: Ubuntu 20.04 + .. tab-item:: ROCm 6.2.1 - Tag - `rocm/tensorflow:rocm6.2.1-py3.9-tf2.16.1-dev `_ (`latest `_) + .. tab-set:: - Inventory - * `ROCm 6.2.1 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.16.1 `_ - * `TensorBoard 2.16.2 `_ + .. tab-item:: TensorFlow 2.16.1 - |hr| + .. tab-set:: - Tag - `rocm/tensorflow:rocm6.2.1-py3.9-tf2.15.1-dev `_ + .. tab-item:: Ubuntu 20.04 - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.15.1 `_ - * `TensorBoard 2.15.2 `_ + Tag + `rocm/tensorflow:rocm6.2.1-py3.9-tf2.16.1-dev `_ (`latest `_) - |hr| + Inventory + * `ROCm 6.2.1 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.16.1 `_ + * `TensorBoard 2.16.2 `_ - Tag - `rocm/tensorflow:rocm6.2.1-py3.9-tf2.14.1-dev `_ + .. tab-item:: TensorFlow 2.15.1 - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.14.1 `_ - * `TensorBoard 2.14.1 `_ + .. tab-set:: - |hr| + .. tab-item:: Ubuntu 20.04 - Tag - `rocm/tensorflow:rocm6.2-py3.9-tf2.16-dev `_ + Tag + `rocm/tensorflow:rocm6.2.1-py3.9-tf2.15.1-dev `_ - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.16.1 `_ - * `TensorBoard 2.16.2 `_ + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.15.1 `_ + * `TensorBoard 2.15.2 `_ - |hr| + .. tab-item:: TensorFlow 2.14.1 - Tag - `rocm/tensorflow:rocm6.2-py3.9-tf2.16-runtime `_ + .. tab-set:: - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.16.1 `_ - * `TensorBoard 2.16.2 `_ + .. tab-item:: Ubuntu 20.04 - |hr| + Tag + `rocm/tensorflow:rocm6.2.1-py3.9-tf2.14.1-dev `_ - Tag - `rocm/tensorflow:rocm6.2-py3.9-tf2.15-dev `_ + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.14.1 `_ + * `TensorBoard 2.14.1 `_ - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.15.1 `_ - * `TensorBoard 2.15.2 `_ + .. tab-item:: ROCm 6.2.0 - |hr| + .. tab-set:: - Tag - `rocm/tensorflow:rocm6.2-py3.9-tf2.15-runtime `_ + .. tab-item:: TensorFlow 2.16.1 - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.15.1 `_ - * `TensorBoard 2.15.2 `_ + .. tab-set:: - |hr| + .. tab-item:: Ubuntu 20.04 - Tag - `rocm/tensorflow:rocm6.2-py3.9-tf2.14-dev `_ + Tag + `rocm/tensorflow:rocm6.2-py3.9-tf2.16-dev `_ - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.14.1 `_ - * `TensorBoard 2.14.1 `_ + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.16.1 `_ + * `TensorBoard 2.16.2 `_ - |hr| + |hr| - Tag - `rocm/tensorflow:rocm6.2-py3.9-tf2.14-runtime `_ + Tag + `rocm/tensorflow:rocm6.2-py3.9-tf2.16-runtime `_ - Inventory - * `ROCm 6.2 `_ - * `Python 3.9 `_ - * `tensorflow-rocm 2.14.1 `_ - * `TensorBoard 2.14.1 `_ + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.16.1 `_ + * `TensorBoard 2.16.2 `_ + + .. tab-item:: TensorFlow 2.15.1 + + .. tab-set:: + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/tensorflow:rocm6.2-py3.9-tf2.15-dev `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.15.1 `_ + * `TensorBoard 2.15.2 `_ + + |hr| + + Tag + `rocm/tensorflow:rocm6.2-py3.9-tf2.15-runtime `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.15.1 `_ + * `TensorBoard 2.15.2 `_ + + .. tab-item:: TensorFlow 2.14.1 + + .. tab-set:: + + .. tab-item:: Ubuntu 20.04 + + Tag + `rocm/tensorflow:rocm6.2-py3.9-tf2.14-dev `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.14.1 `_ + * `TensorBoard 2.14.1 `_ + + |hr| + + Tag + `rocm/tensorflow:rocm6.2-py3.9-tf2.14-runtime `_ + + Inventory + * `ROCm 6.2 `_ + * `Python 3.9 `_ + * `tensorflow-rocm 2.14.1 `_ + * `TensorBoard 2.14.1 `_ .. _install-tensorflow-wheels: diff --git a/docs/sphinx/requirements.txt b/docs/sphinx/requirements.txt index fc5f57db..81d179f5 100644 --- a/docs/sphinx/requirements.txt +++ b/docs/sphinx/requirements.txt @@ -28,7 +28,7 @@ charset-normalizer==3.4.0 # via requests click==8.1.7 # via sphinx-external-toc -cryptography==43.0.3 +cryptography==44.0.0 # via pyjwt defusedxml==0.7.1 # via sphinxcontrib-datatemplates @@ -41,7 +41,7 @@ docutils==0.21.2 # pydata-sphinx-theme # sphinx # sphinx-substitution-extensions -fastjsonschema==2.20.0 +fastjsonschema==2.21.1 # via rocm-docs-core gitdb==4.0.11 # via gitpython @@ -82,7 +82,7 @@ pygments==2.18.0 # accessible-pygments # pydata-sphinx-theme # sphinx -pyjwt[crypto]==2.10.0 +pyjwt[crypto]==2.10.1 # via pygithub pynacl==1.5.0 # via pygithub @@ -146,7 +146,7 @@ sphinxcontrib-runcmd==0.2.0 # via sphinxcontrib-datatemplates sphinxcontrib-serializinghtml==2.0.0 # via sphinx -tomli==2.1.0 +tomli==2.2.1 # via sphinx typing-extensions==4.12.2 # via