Installing ROCm and deep learning frameworks

Installing ROCm and deep learning frameworks#

2025-08-22

3 min read time

Applies to Linux

Before getting started, install ROCm and supported deep learning frameworks.

Pre-install

Each release of ROCm supports specific hardware and software configurations. Before installing, consult the System requirements and Installation prerequisites guides.

If you’re new to ROCm, refer to the ROCm quick start install guide for Linux.

If you’re using a Radeon GPU for graphics-accelerated applications, refer to the Radeon installation instructions.

You can install ROCm on compatible systems via your Linux distribution’s package manager. See the following documentation resources to get started:

Post-install

Follow the post-installation instructions to configure your system linker, PATH, and verify the installation.

If you encounter any issues during installation, refer to the Installation troubleshooting guide.

Deep learning frameworks#

ROCm supports deep learning frameworks and libraries including PyTorch, TensorFlow, JAX, and more.

Review the framework installation documentation. For ease-of-use, it’s recommended to use official ROCm prebuilt Docker images with the framework pre-installed.

Next steps#

After installing ROCm and your desired ML libraries – and before running AI workloads – conduct system health benchmarks to test the optimal performance of your AMD hardware. See System health benchmarks to get started.