How do I specify the use of a particular GPU for inference?

facebookresearch / nougat

Implementation of Nougat Neural Optical Understanding for Academic Documents

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

MIT License

8.81k stars 560 forks source link

Open tfsxyz opened 10 months ago

tfsxyz commented 10 months ago

I have several GPUs, but my GPU0 is underperforming, and I would like to use GPU1. Thank you.

lukas-blecher commented 10 months ago

Try setting the environment variable

export CUDA_VISIBLE_DEVICES=1