GradientSurfer / Draw2Img

A simple web UI for interactive text-guided image to image generation, intended for any age and skill level.
MIT License
324 stars 21 forks source link

Draw2Img does not use AMD GPU #9

Open shvedes opened 5 months ago

shvedes commented 5 months ago

It seems that for some reason all the image generation is using only the CPU instead of the GPU, I have no idea why.

OS: Arch Linux CPU: 5800X GPU: 6700XT

GradientSurfer commented 5 months ago

Hi @shvedes thank you for the report. I don't have an AMD GPU to test with so I won't be much help unfortunately, but I can try to point you in the right direction.

At minimum I believe you'll need to have installed the GPU driver, ROCm, and a pytorch build with ROCm support. I found this post on Reddit which you may find helpful https://www.reddit.com/r/LocalLLaMA/comments/18ourt4/my_setup_for_using_rocm_with_rx_6700xt_gpu_on/.

Please feel free to let me know if there are any changes I can make to this project to better support your environment.

shvedes commented 5 months ago

Tnanks for your reply. I will give a try later and will let you any news