Update Dockerfile

Author: danielckv

Dockerfile

@@ -1,23 +1,26 @@
-# Base image with CUDA 12.6 and Ubuntu 24.04 (includes Python 3.12)
+# Base image with CUDA 12.6 and Ubuntu 24.04
 FROM nvidia/cuda:12.6.0-devel-ubuntu24.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONDONTWRITEBYTECODE=1
+# Fix for Ubuntu 24.04 PEP 668 (allows global pip install)
+ENV PIP_BREAK_SYSTEM_PACKAGES=1 
 ENV TORCH_CUDA_ARCH_LIST="8.0;8.6;8.9;9.0"
 
 # Set the working directory
 WORKDIR /workspace
 
 # 1. Install System Dependencies
+# CHANGED: 'libgl1-mesa-glx' -> 'libgl1' (Fixes exit code 100)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     wget \
     curl \
     vim \
     ffmpeg \
-    libgl1-mesa-glx \
+    libgl1 \
     libglib2.0-0 \
     python3-pip \
     python3-dev \
@@ -28,7 +31,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN ln -s /usr/bin/python3 /usr/bin/python
 
 # 3. Install PyTorch 2.7.0 with PINNED dependencies
-# We pin torchvision==0.22.0 and torchaudio==2.7.0 to prevent conflicts
+# Pinning versions ensures compatibility with PyTorch 2.7
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
     pip install --no-cache-dir \
     torch==2.7.0 \