rocm/pytorch repository overview

This repo hosts the docker images for PyTorch releases with ROCm backend support.

(1) Supported GPUs:

The docker images hosted in this registry will run on:

• gfx908 (MI100)
• gfx90a (MI210/MI250/MI250x)
• gfx942 (MI300A/MI300X/MI325)
• (as of ROCm 7.0) gfx950 (MI350/MI355)
• gfx1030 (Navi21-based SKUs)
• gfx1100/gfx1101 (Navi31-based SKUs)
• (as of ROCm 6.4.2) gfx1200/gfx1201 (Navi44/Navi48-based SKUs)
• (as of ROCm 7.1) gfx1151/gfx1150 (Strix Halo/Strix Point-based SKUs)

(2) Install Docker Environment

To configure docker environment for ROCm, please refer to the following installation guide: https://github.com/RadeonOpenCompute/ROCm-docker/blob/master/quick-start.md⁠

(3) Run the rocm/pytorch Docker Image:

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:​

drun rocm/pytorch​

or

drun rocm/pytorch​:latest