Update Dockerfile

Dockerfile

@@ -1,57 +1,58 @@
 FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
 
-# NVIDIA Container Runtime Configuration
-ENV NVIDIA_VISIBLE_DEVICES=all \
-    NVIDIA_DRIVER_CAPABILITIES=compute,utility \
+# System-level configuration
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONUNBUFFERED=1 \
     CUDA_HOME=/usr/local/cuda \
-    PATH=${CUDA_HOME}/bin:${PATH} \
-    LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
+    PATH=/usr/local/cuda/bin:$PATH \
+    LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH \
     TORCH_CUDA_ARCH_LIST="8.6" \
-    FORCE_CUDA="1"
-
-WORKDIR /app
-
-# System Dependencies with Explicit Versions
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    build-essential \
-    python3-dev \
-    git \
-    ffmpeg \
-    libsndfile1 \
-    curl \
+    NVIDIA_VISIBLE_DEVICES=all \
+    NVIDIA_DRIVER_CAPABILITIES=compute,utility
+
+# System dependencies with explicit version control
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential=12.9ubuntu3 \
+    python3-dev=3.10.6-1~22.04 \
+    python3-pip=22.0.2+dfsg-1ubuntu0.4 \
+    git=1:2.34.1-1ubuntu1.10 \
+    ffmpeg=7:4.4.2-0ubuntu0.22.04.1 \
+    libsndfile1=1.0.31-2ubuntu0.1 \
+    curl=7.81.0-1ubuntu1.15 \
+    && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# Python Environment Setup
-RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
+# Python environment configuration
+RUN ln -sf /usr/bin/python3 /usr/bin/python && \
+    curl -sS https://bootstrap.pypa.io/get-pip.py | python3 && \
+    python3 -m pip install --no-cache-dir pip==23.3.1 setuptools==69.0.3 wheel==0.42.0
 
-# Cache Directory Configuration
-RUN mkdir -p /app/cache && \
-    chmod 777 /app/cache && \
-    mkdir -p /app/src
+# Application directory structure
+WORKDIR /app
+RUN mkdir -p /app/cache /app/src && \
+    chmod 777 /app/cache
 
-# Application Setup - Two-Stage Copy for Better Caching
+# Dependencies installation (two-stage for better caching)
 COPY requirements.txt /app/src/
 WORKDIR /app/src
-RUN pip install --no-cache-dir -r requirements.txt
+RUN python3 -m pip install --no-cache-dir -r requirements.txt
 
-# Copy Application Code
+# Application code
 COPY . /app/src/
 
-# Environment Configuration
-ENV PYTHONUNBUFFERED=1 \
-    MODEL_PATH=/app/cache \
+# Runtime environment
+ENV MODEL_PATH=/app/cache \
     PYTHONPATH=/app/src:$PYTHONPATH \
     OMP_NUM_THREADS=1 \
     MKL_NUM_THREADS=1
 
-# GPU Verification with Detailed Output
-RUN python3 -c "import torch; \
-    assert torch.cuda.is_available(), 'CUDA not available'; \
-    device = torch.cuda.current_device(); \
-    print(f'CUDA Device Index: {device}'); \
-    print(f'CUDA Device Name: {torch.cuda.get_device_name(device)}'); \
-    print(f'CUDA Device Arch: {torch.cuda.get_device_capability(device)}'); \
-    print(f'CUDA Version: {torch.version.cuda}')"
+# GPU verification with comprehensive diagnostics
+RUN python3 -c 'import torch; \
+    assert torch.cuda.is_available(), "CUDA unavailable"; \
+    print(f"PyTorch: {torch.__version__}"); \
+    print(f"CUDA: {torch.version.cuda}"); \
+    print(f"GPU: {torch.cuda.get_device_name()}"); \
+    print(f"Arch: {torch.cuda.get_device_capability()}");'
 
 EXPOSE 8000