search

ckkelvinchan / BasicVSR_PlusPlus

Official repository of "BasicVSR++: Improving Video Super-Resolution with Enhanced Propagation and Alignment"
Apache License 2.0
589 stars 64 forks source link

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

Open PatrickFitzgeraldt opened 1 year ago

PatrickFitzgeraldt commented 1 year ago

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 commented 1 year ago

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 commented 1 year ago

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 .