search

kermitt2 / grobid

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

hold cuda11.2 from upgrading to cuda12.2 #1173

Closed vipulg13 closed 1 month ago

vipulg13 commented 1 month ago

This PR fixes an issue where apt-get upgrade was upgrading CUDA from 11.2 to 12.2, causing mismatches between libcublas 11.2, libcudnn8, and other GPU libraries. This led to GPU-related errors in production.

Changes: Held back library libcudnn8 to prevent it from upgrading to CUDA 12.2.

lfoppiano commented 1 month ago

I tested the resulting image and it works fine.