Multiple GPUs execution

[hellcatmon]

Hello,

I have a system with two GPUs and am running the enhancer in parallel mode. However, I've noticed that the second GPU is not being utilized. Is there anything else I can do to enable execution on both GPUs?

!resemble-enhance --tau 0.48 --nfe 68 --parallel_mode input_folder output_folder

Best Regard,
Igor

[Cooperos]

Hello,

I have a system with two GPUs and am running the enhancer in parallel mode. However, I've noticed that the second GPU is not being utilized. Is there anything else I can do to enable execution on both GPUs?

!resemble-enhance --tau 0.48 --nfe 68 --parallel_mode input_folder output_folder

Best Regard, Igor

Hello, Igor!

Just created a PR. Could you please test it and confirm if your problem is solved?
It would be great if you could provide more information about your hardware if the fix didn't help.

Best Regard, Mikhail

@hellcatmon

[hellcatmon]

Hi Mikhail,

Thank you for the quick update. I've tried the code with a dual T4 GPU setup, but unfortunately, only the first GPU is being utilized.

Here is the command I used, to which I've just added the --distributed_mode parameter:
!resemble-enhance --tau 0.48 --nfe 68 --parallel_mode --distributed_mode input_folder output_folder

Best Regards, Igor