Update Dockerfile

Dockerfile

@@ -3,9 +3,11 @@ FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install Python 3.11 explicitly
-RUN apt-get update && apt-get install -y \
-    python3.11 python3.11-dev python3.11-venv python3-pip \
+# 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 \
     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 && \