[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