Slow Execution of test.py Despite Model and Data Being on cuda:0

facebookresearch / nougat

Implementation of Nougat Neural Optical Understanding for Academic Documents

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

MIT License

8.98k stars 567 forks source link

Slow Execution of test.py Despite Model and Data Being on cuda:0 #184

Open Whalesong-zrs opened 11 months ago

Whalesong-zrs commented 11 months ago

When running test.py, I am experiencing significantly slow performance. I have verified that both the data and the model are on cuda:0. However, I observed that the GPU memory on cuda:0 is not being utilized as expected. Could this be a configuration issue, or is there something else I might be missing?