How to specify which GPU to use?

facebookresearch / nougat

Implementation of Nougat Neural Optical Understanding for Academic Documents

https://facebookresearch.github.io/nougat/

MIT License

8.81k stars 561 forks source link

Closed youyouye closed 12 months ago

youyouye commented 12 months ago

I have multiple GPUs, how can I specify which GPU to use?

lukas-blecher commented 12 months ago

set the environment variable CUDA_VISIBLE_DEVICES to the gpu index you want to use, e.g.

export CUDA_VISIBLE_DEVICES=1

and then call nougat normally