Update Dockerfile

laurynas · Dec 14, 2024 · b93e4fc · b93e4fc
1 parent f524486
commit b93e4fc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,10 @@
 FROM python:3.12-slim
 
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    libcublas-12-0
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
 WORKDIR /app
 
 ADD requirements.txt /app