Update Dockerfile

Dockerfile

@@ -1,20 +1,28 @@
-# Use PyTorch 2.1.0 with CUDA 11.8 (devel version)
-FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
+# Use Ubuntu 22.04 as base (Python 3.11 is default)
+FROM ubuntu:22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install prerequisites and add PPA for Python 3.11
-RUN apt-get update && apt-get install -y software-properties-common && \
-    add-apt-repository -y ppa:deadsnakes/ppa && \
-    apt-get update && apt-get install -y \
-    python3.11 python3.11-dev python3.11-venv python3.11-distutils python3-pip && \
-    update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1 && \
-    update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    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
 
 # Verify Python version
 RUN python --version
 
-# Install a compatible dlib wheel
+# Install CUDA and cuDNN (for PyTorch)
+RUN apt-get install -y \
+    cuda-11-8 \
+    libcudnn8=8.9.2.26-1+cuda11.8
+
+# Install PyTorch (CUDA 11.8)
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+
+# Install a compatible dlib wheel (manylinux_x86_64)
 RUN pip install --no-cache-dir \
     https://www.wheelodex.org/projects/dlib-bin/wheels/dlib_bin-19.24.6-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl