[feat] Support a ROCm image

[benniekiss]

The cuda image has been really helpful for development, but I've recently found a need for rocm development. It would be helpful if a rocm image was added which included the necessary libraries, similar to the cuda image.

[wholtz]

Hi @benniekiss . Can you suggest an image we can use as a starting point? Ideally it would be fairly lightweight and not contain software that is installable from conda packages.

[benniekiss]

Definitely! Part of the reason for my request was because I could not find clear information on the available rocm images, but there are some that appear to be just os + rocm packages, like this ubuntu one. I think it may be built from this repo, but I'm not sure.

[benniekiss]

I was looking more into containerizing the rocm packages, and I have not found a straight forward way to reduce image sizes, which can exceed 50GB. These are quite heavy images! Some comments on rocm repo issues point to efforts to reduce the package size. There's this issue in particular ROCm/ROCm#4224

[wholtz]

If we were to add a ROCm base image, what versions of ROCm would we need to support? Or maybe another way of saying it, does ROCm use semantic versioning such that we only need to have the newest image within each major version number?

[benniekiss]

Here is a list of rocm version history. It appears they use semantic versioning, and the ubuntu image I linked above is also tagged with the relative rocm version. The rocm installation guide also covers multi-versioned and single version installs, but that may be less relevant for containerization