ROCm Docker

Like NVIDIA, AMD also provides Docker usage. It can be installed through four simple steps:

1. Clone the ROCm-docker repository from GitHub (assuming git is installed):

git clone https://github.com/RadeonOpenCompute/ROCm-docker

2. Change the current directory to the newly downloaded ROCm-docker directory:

cd ROCm-docker

3. Next, build rocm-terminal:

sudo docker build -t rocm/rocm-terminal rocm-terminal

4. Finally, add it to the video group:

sudo docker run -it --device=/dev/kfd --device=/dev/dri --group-add video rocm/rocm-terminal

A Docker configuration can also be very easily set up on the PyCharm Professional Edition IDE.