Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Use Ubuntu 22.04 as base (Python 3.11 is default)
-FROM ubuntu:22.04
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 
@@ -14,14 +14,6 @@ RUN apt-get update && apt-get install -y \
 # Verify Python version
 RUN python --version
 
-# 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