kermitt2 / grobid

A machine learning software for extracting information from scholarly documents
https://grobid.readthedocs.io
Apache License 2.0
3.6k stars 460 forks source link

Container 0.8.1 not working with Nvidia GPUs #1170

Closed dcfidalgo closed 2 months ago

dcfidalgo commented 2 months ago

Hi there! Thanks for this super handy tool!! Quick question: did someone manage to use GPUs with the new docker 0.8.1 image? I get the error

Could not load library libcublasLt.so.12. Error: libcublasLt.so.12: cannot open shared object file: No such file or directory

Using version 0.8.0 everything works as expected.

lfoppiano commented 2 months ago

Hi @dcfidalgo thanks for reporting this issue. I've fixed and pushed a new image lfoppiano/grobid:0.8.1-fixes, please check it out. We'll roll out the new updated docker images ASAP.

lfoppiano commented 2 months ago

All docker image have been updated. Sorry for the inconvenience.

dcfidalgo commented 2 months ago

Thanks a lot for the quick fix!!