I’ve used ROCm for pytorch and stable diffusion. Using the pre-built docker images is definitely the way to go. Once setup, it’s worked great for me.
You just have to understand that most projects are going to assume Nvidia (CUDA), so you’ll have to jump through more hoops and still may not be able to do everything.
I’ve used ROCm for pytorch and stable diffusion. Using the pre-built docker images is definitely the way to go. Once setup, it’s worked great for me.
You just have to understand that most projects are going to assume Nvidia (CUDA), so you’ll have to jump through more hoops and still may not be able to do everything.