2 ms·

I've had more luck with the ROCm docker container. I run it via k8s. It was pretty painless to set up and has been mostly painless since. Prior to that it was n

I've had more luck with the ROCm docker container. I run it via k8s. It was pretty painless to set up and has been mostly painless since. Prior to that it was nearly impossible to get Jax running reliably on ROCm. Even with the container, you have to be careful installing Python libraries because they can still break things.

lenova 2y ago
I just recently went down the AMD GPU + ROCm rabbit hole as well. ROCm 6.2 was just released in August of this year and introduces a lot better support, though as the above poster mentioned, isn't merged into most recent OSes. This Github repo is good for tracking the latest Ubuntu + ROCm install process: https://github.com/nktice/AMD-AI https://github.com/nktice/AMD-AI
- latchkey 2y ago
  That's a nice repo of random installation notes. Very helpful, thanks!