rocm/cupy repository overview

This repo hosts the docker images for CuPy releases with ROCm backend support. To configure docker environment for ROCm, please refer to the following installation guide: https://github.com/RadeonOpenCompute/ROCm-docker/blob/master/quick-start.md⁠

The docker images been hosted in this registry will run on gfx906 (Vega20-type GPU - MI50, MI60), gfx908 (MI100) and gfx90a (MI210/MI250/MI250x).

Starting from ROCm5.5, the "latest" tag docker images also contain development support on CuPy.

Commands to pull and run the latest CuPy docker container. Set environment variable alias:

alias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --ipc=host --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --shm-size=64G -v $HOME/dockerx:/dockerx'

Run command:

drun rocm/cupy