mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-04 03:15:28 -05:00
100 lines
6.0 KiB
ReStructuredText
100 lines
6.0 KiB
ReStructuredText
******************************************************************
|
|
Docker image support matrix
|
|
******************************************************************
|
|
|
|
ROCm 5.7 supports the following `Pytorch <https://hub.docker.com/r/rocm/pytorch>`_ and
|
|
`TensorFlow <https://hub.docker.com/r/rocm/tensorflow>`_ Docker images:.
|
|
|
|
|
|
.. tab-set::
|
|
|
|
.. tab-item:: PyTorch
|
|
|
|
.. tab-set::
|
|
|
|
.. tab-item:: Ubuntu
|
|
|
|
* ``rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_staging``
|
|
|
|
* `ROCm 5.7 <https://repo.radeon.com/rocm/apt/5.7/>`_
|
|
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
|
|
* `Torch 2.1.0 <https://github.com/ROCmSoftwarePlatform/pytorch/tree/rocm5.7_internal_testing>`_
|
|
* `Apex 0.1 <https://github.com/ROCmSoftwarePlatform/apex/tree/v0.1>`_
|
|
* `Torchvision 0.16.0 <https://github.com/pytorch/vision/tree/release/0.16>`_
|
|
* `Tensorboard 2.14.0 <https://github.com/tensorflow/tensorboard/tree/2.14>`_
|
|
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
|
|
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
|
|
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
|
|
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_
|
|
|
|
|
|
* ``Ubuntu rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_1.12.1``
|
|
|
|
* `ROCm 5.7 <https://repo.radeon.com/rocm/apt/5.7/>`_
|
|
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
|
|
* `Torch 1.12.1 <https://github.com/ROCmSoftwarePlatform/pytorch/tree/release/1.12>`_
|
|
* `Apex 0.1 <https://github.com/ROCmSoftwarePlatform/apex/tree/v0.1>`_
|
|
* `Torchvision 0.13.1 <https://github.com/pytorch/vision/tree/v0.13.1>`_
|
|
* `Tensorboard 2.14.0 <https://github.com/tensorflow/tensorboard/tree/2.14>`_
|
|
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
|
|
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
|
|
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
|
|
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_
|
|
|
|
* ``Ubuntu rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_1.13.1``
|
|
|
|
* `ROCm 5.7 <https://repo.radeon.com/rocm/apt/5.7/>`_
|
|
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
|
|
* `Torch 1.12.1 <https://github.com/ROCmSoftwarePlatform/pytorch/tree/release/1.13>`_
|
|
* `Apex 0.1 <https://github.com/ROCmSoftwarePlatform/apex/tree/v0.1>`_
|
|
* `Torchvision 0.14.0 <https://github.com/pytorch/vision/tree/v0.14.0>`_
|
|
* `Tensorboard 2.12.0 <https://github.com/tensorflow/tensorboard/tree/2.12.0>`_
|
|
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
|
|
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
|
|
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
|
|
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_
|
|
|
|
* ``Ubuntu rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_2.0.1``
|
|
|
|
* `ROCm 5.7 <https://repo.radeon.com/rocm/apt/5.7/>`_
|
|
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
|
|
* `Torch 2.0.1 <https://github.com/ROCmSoftwarePlatform/pytorch/tree/release/2.0>`_
|
|
* `Apex 0.1 <https://github.com/ROCmSoftwarePlatform/apex/tree/v0.1>`_
|
|
* `Torchvision 0.15.2 <https://github.com/pytorch/vision/tree/release/0.15>`_
|
|
* `Tensorboard 2.14.0 <https://github.com/tensorflow/tensorboard/tree/2.14>`_
|
|
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
|
|
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
|
|
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
|
|
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_
|
|
|
|
.. tab-item:: CentOS 7
|
|
|
|
``rocm/pytorch:rocm5.7_centos7_py3.9_pytorch_staging``
|
|
|
|
* `ROCm 5.7 <https://repo.radeon.com/rocm/yum/5.7/>`_
|
|
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
|
|
* `Torch 2.1.0 <https://github.com/ROCmSoftwarePlatform/pytorch/tree/rocm5.7_internal_testing>`_
|
|
* `Apex 0.1 <https://github.com/ROCmSoftwarePlatform/apex/tree/v0.1>`_
|
|
* `Torchvision 0.16.0 <https://github.com/pytorch/vision/tree/release/0.16>`_
|
|
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
|
|
|
|
.. tab-item:: TensorFlow
|
|
|
|
.. tab-set::
|
|
|
|
.. tab-item:: Ubuntu
|
|
|
|
* ``rocm5.7_ubuntu20.04_py3_tensorflow_r2.12-rocm-enhanced_release``
|
|
|
|
* `ROCm 5.7 <https://repo.radeon.com/rocm/apt/5.7/>`_
|
|
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
|
|
* `tensorflow-rocm 2.12.1 <https://pypi.org/project/tensorflow-rocm/2.12.1.570/>`_
|
|
* `Tensorboard 2.12.3 <https://github.com/tensorflow/tensorboard/tree/2.12>`_
|
|
|
|
* ``rocm5.7_ubuntu20.04_py3_tensorflow_r2.13-rocm-enhanced_release``
|
|
|
|
* `ROCm 5.7 <https://repo.radeon.com/rocm/apt/5.7/>`_
|
|
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
|
|
* `tensorflow-rocm 2.13.0 <https://pypi.org/project/tensorflow-rocm/2.13.0.570/>`_
|
|
* `Tensorboard 2.13.0 <https://github.com/tensorflow/tensorboard/tree/2.13>`_
|