Open dingyan20 opened 4 months ago
Hi,
It seems the model runs on only one GPU by default. Is it possible to run it on multiple GPUs. My machine has 4 GPUs. But the memory of each is only about 10 G, which is not enough to run the model.
Thank you very much!
You can simply divide the task into four parts, and then add the corresponding GPU number before Python to fully utilize all the GPUs.
Hi,
It seems the model runs on only one GPU by default. Is it possible to run it on multiple GPUs. My machine has 4 GPUs. But the memory of each is only about 10 G, which is not enough to run the model.
Thank you very much!