Sounds really awesome. Any chance someone can
suggest if this works also inside a Docker container?
https://rocm.docs.amd.com/projects/install-on-linux/en/lates...
There are some delightful AMD driver issues that make certain models of GPU intermittently freeze the kernel when used from docker. That was great fun when building SCALE's CI system :D.