facebookresearch / nougat

Implementation of Nougat Neural Optical Understanding for Academic Documents
https://facebookresearch.github.io/nougat/
MIT License
8.95k stars 567 forks source link

can nougat utilize multi GPUs? #63

Open jackgreen12345 opened 1 year ago

jackgreen12345 commented 1 year ago

in nougat's options, we don't need to allocate GPUs. we have multiple GPUs in the server, can nougat use multi GPUs in one task? if nougat can only use one GPU in one task, can we run serveral tasks which one task run with one GPU?

lukas-blecher commented 1 year ago

No, inference only works on 1 GPU. You'll have to split the load manually, as you suggested.

marwinsteiner commented 1 year ago

in nougat's options, we don't need to allocate GPUs. we have multiple GPUs in the server, can nougat use multi GPUs in one task? if nougat can only use one GPU in one task, can we run serveral tasks which one task run with one GPU?

have you already implemented that? Would be interested in seeing your approach to this