asfads

Dockerfile

@@ -1,5 +1,6 @@
 FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
 
+
 # 1) Variables HF avant tout
 ENV HF_HOME="/home/user/.cache/huggingface" \
     HF_HUB_CACHE="/home/user/.cache/huggingface/hub" \
@@ -8,6 +9,11 @@ ENV HF_HOME="/home/user/.cache/huggingface" \
 # 2) Créer l’utilisateur non-root
 RUN useradd -m -u 1000 user
 
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+      python3 python3-pip python3-dev && \
+    rm -rf /var/lib/apt/lists/*
+
 # 3) Installer dépendances système
 RUN apt-get update && apt-get install -y \
       build-essential git libpoppler-cpp-dev poppler-utils libmagic-dev python3-dev \
@@ -16,7 +22,7 @@ RUN apt-get update && apt-get install -y \
 # 4) Copier requirements et installer libs Python
 WORKDIR /home/user/app
 COPY --chown=user requirements.txt .
-RUN pip install --no-cache-dir torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir torch torchvision --extra-index-url https://download.pytorch.org/whl/cu118
 RUN pip install --no-cache-dir huggingface_hub
 RUN pip install --no-cache-dir -r requirements.txt
 

requirements.txt

@@ -1,5 +1,8 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.7.0
+
+torch==2.2.2+cu118
+torchvision==0.17.2+cu118
+torchaudio==2.2.2+cu118
 annotated-types==0.7.0
 attrs==25.3.0
 beautifulsoup4==4.13.4