Update Dockerfile

Dockerfile

@@ -1,15 +1,22 @@
-# Use a newer PyTorch image with Python 3.11
-FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+# Use PyTorch 2.1.0 with CUDA 11.8 (devel version)
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install system dependencies
+# Install Python 3.11 explicitly
 RUN apt-get update && apt-get install -y \
-    git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev build-essential cmake \
-    libboost-all-dev python3-dev
+    python3.11 python3.11-dev python3.11-venv python3-pip \
+    git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev \
+    build-essential cmake libboost-all-dev && \
+    ln -sf /usr/bin/python3.11 /usr/bin/python && \
+    ln -sf /usr/bin/pip3 /usr/bin/pip
 
-# Install dlib via pip (Python 3.11 should be available now)
-RUN pip install --no-cache-dir dlib
+# Confirm Python version
+RUN python --version
+
+# Use a prebuilt dlib wheel instead of compiling
+RUN pip install --no-cache-dir \
+    https://www.piwheels.org/simple/dlib/dlib-19.24.6-cp311-cp311-linux_x86_64.whl
 
 # Set up user
 RUN useradd -m -u 1000 user