dffdfddf

Dockerfile

@@ -1,17 +1,15 @@
-FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install base dependencies
+# Install Python and ML dependencies
 RUN apt-get update && \
-    apt-get install -y software-properties-common && \
-    add-apt-repository ppa:deadsnakes/ppa && \
-    apt-get update && \
     apt-get install -y \
         python3.10 \
         python3.10-venv \
         python3.10-distutils \
         python3-pip \
+        python3-dev \
         wget \
         git \
         libgl1 \
@@ -40,7 +38,7 @@ WORKDIR /app
 COPY --chown=user:user requirements.txt .
 
 # Upgrade pip and install PyTorch dependencies first
-# Use versions compatible with CUDA 12.1 and L40S
+# Use versions compatible with CUDA 12.1 for NVIDIA L4 support
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir \
         torch==2.1.2 \