Date: 23.08.2025

AMD ROCm Docker Images for PyTorch / TensorFlow

A concise snapshot of the latest minor/patch releases per major ROCm line (starting from ROCm 5), mapped to corresponding PyTorch/TensorFlow images.

• Only fixed tags (no latest).
• Images are the official rocm/pytorch and rocm/tensorflow images.
• Base OS is included when the tag or AMD docs explicitly indicate it; in some TF tags it is inferred from the Python version (e.g., py3.12 → Ubuntu 24.04; py3.10 → Ubuntu 22.04).

ROCm version	Base OS	Image	Framework version
6.4.3	Ubuntu 24.04	rocm/pytorch:rocm6.4.3_ubuntu24.04_py3.12_pytorch_release_2.6.0	PyTorch 2.6.0
6.4.3	Ubuntu 22.04	rocm/pytorch:rocm6.4.3_ubuntu22.04_py3.10_pytorch_release_2.5.1	PyTorch 2.5.1
6.4.2	Ubuntu 24.04	rocm/tensorflow:rocm6.4.2-py3.12-tf2.18-dev	TensorFlow 2.18.1
6.4.2	Ubuntu 22.04	rocm/tensorflow:rocm6.4.2-py3.10-tf2.18-dev	TensorFlow 2.18.1
6.3.4	Ubuntu 22.04	rocm/pytorch:rocm6.3.4_ubuntu22.04_py3.10_pytorch_release_2.4.0	PyTorch 2.4.0
6.3.1	Ubuntu 24.04	rocm/tensorflow:rocm6.3.1-py3.12-tf2.17.0-dev	TensorFlow 2.17.0
6.3.1	Ubuntu 22.04	rocm/tensorflow:rocm6.3.1-py3.10-tf2.17.0-dev	TensorFlow 2.17.0
6.2.2	Ubuntu 22.04	rocm/pytorch:rocm6.2.2_ubuntu22.04_py3.10_pytorch_release_2.3.0	PyTorch 2.3.0
6.2.2	Ubuntu 22.04*	rocm/tensorflow:rocm6.2.2-py3.9-tf2.16.2-dev	TensorFlow 2.16.2
6.1.2	Ubuntu 22.04	rocm/pytorch:rocm6.1.2_ubuntu22.04_py3.10_pytorch_release-2.1.2	PyTorch 2.1.2
6.1.x	Ubuntu 22.04*	rocm/tensorflow:rocm6.1-py3.9-tf2.14-runtime	TensorFlow 2.14.0
6.0.2	Ubuntu 22.04	rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2	PyTorch 2.1.2
6.0.x	Ubuntu 22.04*	rocm/tensorflow:rocm6.0-tf2.13-dev	TensorFlow 2.13.x
5.7	Ubuntu 22.04	rocm/pytorch:rocm5.7_ubuntu22.04_py3.10_pytorch_2.0.1	PyTorch 2.0.1
5.5	Ubuntu 20.04*	rocm/tensorflow:rocm5.5-tf2.10-dev	TensorFlow 2.10

---

Notes

• OS inference: In TF tags where the OS is not encoded, OS is inferred from Python version or time period; entries marked with * are best-effort.
• For TF 6.4.x, AMD documentation explicitly lists Ubuntu 24.04 for py3.12 and Ubuntu 22.04 for py3.10 tags.