Can BasicVSR++ Inference a video with multi-GPUs ?

[PatrickFitzgeraldt]

First of all, thanks for your great work. BasicVSR++ is the best denoising model I could find.
I have a workstation with 4 3090, but when I tried to run the code on my own machine. I noticed that there is only one GPU was working.
Will BasicVSR++ support Inferenceing a video with multi-GPUs?
thank you.

[CarlGao4]

I'm currently writing a new demo with optimized memory use. I'll try to add multi-GPU support if possible.
You can view it at https://github.com/CarlGao4/BasicVSR_PlusPlus/blob/master/demo/restoration_large_video_demo.py

[PatrickFitzgeraldt]

I'm currently writing a new demo with optimized memory use. I'll try to add multi-GPU support if possible. You can view it at https://github.com/CarlGao4/BasicVSR_PlusPlus/blob/master/demo/restoration_large_video_demo.py

Maybe you can try 🤗 Accelerate ，this might help you add multi-GPU support .