Docker image support matrix#
Applies to Linux
2024-09-30
5 min read time
AMD validates and publishes ready-made PyTorch and TensorFlow images with ROCm backends on Docker Hub. The following Docker image tags and associated inventories are validated for ROCm 6.2.
Refer to Using a Docker image with PyTorch pre-installed and Using a prebuilt Docker image for installation instructions.
- 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:latestpoints to a Docker image with the latest ROCm tested release version of PyTorch (for example, version 2.3), similar torocm/pytorch:latest-releasetag. See Using a Docker image with PyTorch pre-installed for more information.- Inventory
- Tag
- Inventory
- Tag
- Inventory
- Tag
- Inventory
- Tag
- Inventory
- Tag
- Inventory
- Tag
- Inventory
- Tag
- Inventory
- Tag
- Inventory
- Tag
- Inventory