AMD GPU Support?

#7
by Viper9978 - opened

Is it possible to run this with an AMD GPU? I followed the install instructions and it's stuck at 0%. I installed the rocm version of pytorch and changed my device ID to match my 7900xtx and it's not generating anything.

i have it running using comfyui-Zlulda on a 6950

ACE-Step org

We currently don’t have any modern consumer-grade AMD GPUs available for testing. You may want to wait for further updates from the open-source community.

That's a shame, I would also like AMD support.

I have a succesful torch ROCM setup.
(ace_step) user@host:~/ACE-Step$ python3 -c 'import torch; print(torch.cuda.is_available())'
True

But when launching acestep, i get the following error:
OSError: libtorch_cuda.so: cannot open shared object file: No such file or directory

Sign up or log in to comment