Updated Dockerfile for GPU Dependencies

Dockerfile

@@ -6,35 +6,31 @@ ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=$CUDA_HOME/bin:$PATH
 ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
 
-# Check CUDA version
-RUN nvcc --version
-
-# Check NVIDIA driver version
-RUN nvidia-smi
-
 # # 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
 
-# RUN apt-get -y update \
-#     && apt-get install -y software-properties-common
+RUN apt-get -y update \
+    && apt-get install -y software-properties-common
+
+RUN apt-get -y update \
+    && add-apt-repository ppa:deadsnakes/ppa
 
-# RUN apt-get -y update \
-#     && add-apt-repository ppa:deadsnakes/ppa
+RUN apt-get -y update \
+    && apt-get -y install python3.10 \
+    && apt-get -y install python3-pip
 
-# RUN apt-get -y update \
-#     && apt-get -y install python3.10 \
-#     && apt-get -y install python3-pip
+RUN apt-get -y update \
+    && apt-get -y install git
 
-# RUN apt-get -y update \
-#     && apt-get -y install git
+# Additional system dependencies
+RUN apt-get -y update \
+    && apt-get install -y ffmpeg \
+    && apt-get install -y yt-dlp
 
-# # Additional system dependencies
-# RUN apt-get -y update \
-#     && apt-get install -y ffmpeg \
-#     && apt-get install -y yt-dlp
+RUN python --version
 
 # # FROM python:3.10