Installing ROCm and deep learning frameworks
Contents
Installing ROCm and deep learning frameworks#
2025-10-20
3 min read time
Before getting started, install ROCm and supported deep learning frameworks.
Each release of ROCm supports specific hardware and software configurations. Before installing, consult theSystem requirements andInstallation prerequisites guides.
If you’re new to ROCm, refer to theROCm quick start install guide for Linux.
If you’re using a Radeon GPU for graphics-accelerated applications, refer to theRadeon installation instructions.
You can install ROCm oncompatible systems via your Linuxdistribution’s package manager. See the following documentation resources to get started:
Follow thepost-installation instructions toconfigure your system linker, PATH, and verify the installation.
If you encounter any issues during installation, refer to theInstallation troubleshooting guide.
Deep learning frameworks#
ROCm supports deep learning frameworks and libraries includingPyTorch,TensorFlow,JAX, and more.
Review theframework installation documentation. For ease-of-use, it’s recommended to use official ROCm prebuilt Dockerimages with the framework pre-installed.
Next steps#
After installing ROCm and your desired ML libraries – and before running AI workloads – conduct system health benchmarksto test the optimal performance of your AMD hardware. SeeSystem setup for AI workloads on ROCm to get started.