Updated Dockerfile for GPU Dependencies

Dockerfile

@@ -9,13 +9,21 @@ ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
 # Set DEBIAN_FRONTEND to noninteractive to prevent interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install NVIDIA driver
-RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460
+# # Install NVIDIA driver
+# RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460
+
+RUN apt-get -y update \
+    && apt-get install -y software-properties-common \
+    && apt-get -y update \
+    && add-apt-repository universe
+RUN apt-get -y update \
+    && apt-get -y install python3
+    && apt-get -y install python3-pip
 
 # Additional system dependencies
 RUN apt-get install -y ffmpeg && apt-get install -y yt-dlp
 
-FROM python:3.10
+# FROM python:3.10
 
 # Switch back to the root user to install Python packages
 USER root