By default, PyTorch builds for a broad set of AMD architectures. To speed up compilation, you can target only your GPU architecture. ... Replace <uarch> with the result from rocminfo (for example, gfx90a, gfx1030).

March 24, 2021 - An installable Python package is now hosted on pytorch.org, along with instructions for local installation in the same simple, selectable format as PyTorch packages for CPU-only configurations and other GPU platforms. PyTorch on ROCm includes full capability for mixed-precision and large-scale training using AMD’s MIOpen & RCCL libraries.

November 21, 2025 - Determine the architecture (Navi 21, Phoenix, etc.). Use GPU-Z or a similar tool to check. Head to the open driver sources and find the code name for your architecture — for example, Phoenix is gfx1103, Navi 21 is gfx1030.

PYTORCH_TEST_WITH_ROCM=1 python3 test/test_nn.py --verbose · You can replace test_nn.py with any other test set. The PyTorch examples repository provides basic examples that exercise the functionality of your framework.

Tensors and Dynamic neural networks in Python with strong GPU acceleration - computerguy2030/pytorch-rocm-amd

The Instinct MI300X workload optimization guide provides detailed guidance on optimizing workloads for the AMD Instinct MI300X GPU using ROCm. This guide helps users achieve optimal performance for deep learning and other high-performance computing tasks on the MI300X GPU. The Inception with PyTorch documentation describes how PyTorch integrates with ROCm for AI workloads.

To configure docker environment for ROCm, please refer to the following installation guide: https://github.com/RadeonOpenCompute/ROCm-docker/blob/master/quick-start.md⁠ ... To facilitate commands to pull and run the latest PyTorch docker container, add the following alias to your .profile or .bashrc:​ · alias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add=video --ipc=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --shm-size 8G -v $HOME/dockerx:/dockerx -w /dockerx' For example, to get the latest PyTorch on ROCm, run the following command:​

git clone https://github.com/ROCm/MAD cd MAD/scripts/pytorch_train ... The following benchmarking examples require downloading models and datasets from Hugging Face.

April 21, 2026 - This guide describes how to use TrainJob to train or fine-tune AI models with PyTorch on AMD ROCm GPUs on Kubernetes.

PYTORCH_TEST_WITH_ROCM=1 python3 test/test_nn.py --verbose · test_nn.py can be replaced with any other test set. The PyTorch examples repository provides basic examples that exercise the functionality of the framework. MNIST (Modified National Institute of Standards and Technology) database is a collection of handwritten digits that may be used to train a Convolutional Neural Network for handwriting recognition.

PYTORCH_TEST_WITH_ROCM=1 python3 test/run_test.py --verbose

November 15, 2019 - As compiling PyTorch often can ... so just adding ccache on the path won't work for the HIP compilation. The key binary for compilation is /opt/rocm/llvm/bin/clang-11....

Example result: device name [0]: Radeon RX 7900 XTX ... Enter command to display component information within the current PyTorch environment. ... PyTorch version: 2.9.1+rocm7.2.1 Is debug build: False CUDA used to build PyTorch: N/A ROCM used to build PyTorch: 7.2.53211-158bd99533 OS: Microsoft ...

You can run PyTorch code inside of: ---> Intel(R) Core(TM) i5-10600K CPU @ 4.10GHz ---> gfx1012 ... Mint 21.1 (Ubuntu 22.04) rocm works with GPU support on stable-diffusion.

Example: pip3 install numpy==1.26.4 · Select the applicable Ubuntu version and enter the following command to pull the public PyTorch Docker image. Optional: You can also download a specific and supported configuration with different user-space ROCm versions, PyTorch versions, and supported operating systems.

October 15, 2025 - This solution involves maintaining separate python package indices to host the packages for each platform and version. For example AMD ROCm™ Software 6.4-supported packages are hosted at https://download.pytorch.org/whl/rocm6.4.

Example: pip3 install numpy==1.26.4 · Select the applicable Ubuntu version and enter the following command to pull the public PyTorch Docker image. Optional: You can also download a specific and supported configuration with different user-space ROCm versions, PyTorch versions, and supported operating systems.

October 21, 2025 - This applies similarly to ROCm versions, meaning PyTorch releases are tested against ROCm N-1, N, and N+1 versions. This includes minor version. For example, today’s version (N) is ROCm 7.0, N-1 would be 6.4, N+1 would be 7.1.