Nougat ver 0.1.15 & 0.1.16 not utilizing GPU in Colab

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

Nougat ver 0.1.15 & 0.1.16 not utilizing GPU in Colab #137

Closed tsycstang closed 12 months ago

tsycstang commented 12 months ago

In Google Colab with T4 GPU, newer 0.1.15 & 0.1.16 version do not detect and utilize GPU correctly and fall back to CPU in command line mode, api mode not tested.

My temporary fix is: !pip install nougat-ocr==0.1.14

MereHappyJaZZ commented 12 months ago

Same issue on windows, works by going back to 0.1.14

lukas-blecher commented 12 months ago

Thank you, fixed now.