Change to debian based python image

Dilydigre · May 29, 2024 · 734d026 · 734d026
1 parent 685781d
commit 734d026
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,10 +1,10 @@
-FROM python:3.10-alpine
+FROM python:3.10
 WORKDIR /app
 
 COPY requirements_api.txt .
 COPY requirements_ai.txt .
 
-RUN apk update && apk add git wget cargo musl-dev linux-headers g++ gfortran
+RUN apt update && apt install git wget cargo
 
 RUN pip  install --upgrade pip
 RUN pip install nvidia-pyindex

diff --git a/requirements_ai.txt b/requirements_ai.txt
@@ -62,6 +62,18 @@ multidict==6.0.5
 namex==0.0.8
 networkx==3.3
 numpy==1.26.4
+nvidia-cublas-cu12
+nvidia-cuda-cupti-cu12
+nvidia-cuda-nvrtc-cu12
+nvidia-cuda-runtime-cu12
+nvidia-cudnn-cu12
+nvidia-cufft-cu12
+nvidia-curand-cu12
+nvidia-cusolver-cu12
+nvidia-cusparse-cu12
+nvidia-nccl-cu12
+nvidia-nvjitlink-cu12
+nvidia-nvtx-cu12
 omegaconf==2.3.0
 opencv-python-headless==4.9.0.80
 opt-einsum==3.3.0
@@ -93,7 +105,7 @@ requests==2.31.0
 rich==13.7.1
 rpds-py==0.18.1
 safetensors==0.4.3
-scann
+scann==1.3.2
 scikit-image==0.23.2
 scipy==1.13.1
 shellingham==1.5.4