Install PyTorch on ROCm 7.12.0

This topic guides you through installing PyTorch with ROCm support on AMD hardware. It applies to supported AMD GPUs and platforms.

AMD device family

Instinct

Radeon PRO

Radeon

Ryzen

Instinct GPU

MI355X

MI350X

MI325X

MI300X

MI300A

MI250X

MI250

MI210

MI100

Radeon PRO GPU

AI PRO R9700

AI PRO R9600D

W7900 Dual Slot

W7900

W7800 48GB

W7800

W7700

V710

Radeon GPU

RX 9070 XT

RX 9070 GRE

RX 9070

RX 9060 XT LP

RX 9060 XT

RX 9060

RX 7900 XTX

RX 7900 XT

RX 7900 GRE

RX 7800 XT

RX 7700 XT

RX 7700 XE

RX 7700

RX 7600

Ryzen APU

AI Max+ PRO 395

AI Max PRO 390

AI Max PRO 385

AI Max PRO 380

AI Max+ 395

AI Max 390

AI Max 385

AI 9 HX PRO 475

AI 9 HX PRO 470

AI 9 PRO 465

AI 7 PRO 450

AI 5 PRO 440

AI 5 PRO 435

AI 9 HX 375

AI 9 HX 370

AI 9 365

9 270

7 260

7 250

5 240

5 230

5 220

3 210

Operating system

Linux

Operating system

Linux

Operating system

Linux

Windows

Operating system

Linux

Windows

Prerequisites

Ensure your system has a supported Python version installed and accessible: 3.11, 3.12, or 3.13.

Review the ROCm 7.12.0 compatibility matrix for more details.

Install PyTorch

For prerequisite steps and post-installation recommendations, see the ROCm installation instructions.

1. Set up your Python virtual environment. If you already have a successful ROCm 7.12.0 installation using pip, skip this step.

   For example, run the following command to create a virtual environment:

   python3.12 -m venv .venv
   

   py -3.12 -m venv .venv
   

2. Activate your Python virtual environment. For example:

   source .venv/bin/activate
   

   .venv\Scripts\activate
   

3. Install the appropriate ROCm-enabled PyTorch build for your operating system and AMD hardware architecture.

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx950-dcgpu/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx94X-dcgpu/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx90a/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx908/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx120X-all/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx110X-dgpu/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx103X-dgpu/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx1151/ torch torchvision torchaudio
   

   python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx1150/ torch torchvision torchaudio
   

4. Check your PyTorch installation.

   python -c "import torch; print(torch.cuda.is_available())"
   

   This prints True if PyTorch and ROCm are installed properly.

Optionally, for a demonstration of ROCm-powered image generation, try ComfyUI.