How to use 2 GPUs for inference

ckkelvinchan / RealBasicVSR

Official repository of "Investigating Tradeoffs in Real-World Video Super-Resolution"

Apache License 2.0

906 stars 136 forks source link

How to use 2 GPUs for inference #48

Open ipeaking opened 2 years ago

ipeaking commented 2 years ago

Hello，I want to use 2 GPUs to inference, But I don't know how to run it in 2 GPUs, thanks.

ckkelvinchan commented 2 years ago

Hello, demo does not support 2GPUs. But you can use multiple GPUs by using dist_test.py with the corresponding config. In particular, you can modify the data paths here and use dist_test.py.