Revert to the cuda runtime version

cublas and cudnn are the issue, not drivers or cuda...
LibrePhotos · Dec 20, 2024 · 6a5c80c · 6a5c80c
1 parent 09de22b
commit 6a5c80c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend-gpu/base/Dockerfile b/backend-gpu/base/Dockerfile
@@ -98,7 +98,7 @@ RUN git clone --depth 1 --branch 'master' https://github.com/davisking/dlib.git
 RUN ls -l /usr/local/lib/python3.12/dist-packages
 
 # Stage 2: Deployment Stage
-FROM ubuntu:noble
+FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04
 ARG TARGETPLATFORM
 ENV DEBIAN_FRONTEND=noninteractive
 WORKDIR /app