Support for AMD ROCm?

[temrb]

I see that this only support NVIDIA gpus on windows. Will there be support for AMD/ i.e. ROCm gpus?

[hacksider]

None of the maintainers have AMD GPU's. So there's no way for us to test it. This is an open source project though so let's wait for the community to make a push on it.

[Oraclemain]

I need help to learn 😫

[Jeepmb]

I've been playing around the last couple days and have found some success replacing onnxruntime-gpu with onnxruntime-rocm (and some various dependencies like torchaudio-rocm) from the AMD repos at https://repo.radeon.com/rocm/manylinux to get GPU accelerated video output on Ubuntu 24.04 using a Radeon RX 7900 XTX. I use the option --execution-provider rocm . This makes the framerate of the webcam at least useable (compared to CPU), however the face swap itself doesn't quite execute correctly, as I just get a distorted/blurred face instead of a clean face swap.

Frankly I've dabbled in this type of stuff for a while but this would be my first time contributing, so I'm not sure what would be the best way to share my findings or collaborate on solutions.

[hacksider]

Hi @Jeepmb , you can just push your code as a PR and we will review it 😄

I'm pretty sure the community will be very glad if they see an improvement for AMD GPU's