GradientSurfer / Draw2Img

A simple web UI for interactive text-guided image to image generation, intended for any age and skill level.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Draw2Img does not use AMD GPU

shvedes opened this issue · comments

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

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.

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