zyddnys / manga-image-translator

Translate manga/image 一键翻译各类图片内文字 https://cotrans.touhou.ai/
https://cotrans.touhou.ai/
GNU General Public License v3.0
4.59k stars 473 forks source link

[Feature Request]: Can multiple GPUs be supported? #655

Closed chou18194766xx closed 1 week ago

chou18194766xx commented 1 week ago

What would your feature do?

It seems that multi-GPU support is not available, or maybe I missed something. How can I enable the multi-GPU working mode? This is quite important on platforms like Kaggle or Colab.

zyddnys commented 1 week ago

do you need multiple GPU because a single one doesn't have enough VRAM?

chou18194766xx commented 1 week ago

do you need multiple GPU because a single one doesn't have enough VRAM?

Thank you for your reply. Most of the time it is sufficient, but sometimes I have to deal with tens of thousands of high-resolution pictures and I want to speed it up. Looking at the idle GPU, I wonder if I can use it and double the processing speed.

zyddnys commented 1 week ago

you can start multiple processes and use CUDA_VISIBLE_DEVICES to select a GPU

chou18194766xx commented 1 week ago

you can start multiple processes and use CUDA_VISIBLE_DEVICES to select a GPU你可以启动多个进程,并使用 CUDA_VISIBLE_DEVICES 来选择 GPU

Thank you