Update Dockerfile

Dockerfile

@@ -1,32 +1,24 @@
 FROM python:3.10-slim
 
-# System libraries
+# System dependencies
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     libgl1 \
     libglib2.0-0 \
     libsm6 \
     libxext6 \
-    fontconfig \
-    fonts-dejavu \
+    g++ \
     && rm -rf /var/lib/apt/lists/*
 
-# Environment configuration
-ENV MPLCONFIGDIR=/tmp/matplotlib \
-    FONTCONFIG_PATH=/tmp/fontconfig \
-    XDG_CACHE_HOME=/tmp \
-    PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.9 \
-    TF_CPP_MIN_LOG_LEVEL=3
-
-# Create writable directories
-RUN mkdir -p ${MPLCONFIGDIR} \
-    && mkdir -p ${FONTCONFIG_PATH} \
-    && chmod -R 777 /tmp
+# Configure environment
+ENV XDG_CACHE_HOME=/tmp \
+    TORCH_USE_CUDA_DSA=1 \
+    PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.9
 
 WORKDIR /app
 COPY . .
 
-# Install dependencies with optimized build
+# Install with optimized dependencies
 RUN pip install --no-cache-dir -U pip && \
     pip install --no-cache-dir -r requirements.txt