Updated Dockerfile for GPU Dependencies

Dockerfile

@@ -6,6 +6,15 @@ ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=$CUDA_HOME/bin:$PATH
 ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
 
+# Install necessary utilities (e.g., for querying versions)
+RUN apt-get update \
+    && apt-get install -y nvidia-smi \
+    && rm -rf /var/lib/apt/lists/*
+
+# Check CUDA version
+RUN nvcc --version
+
+# Check NVIDIA driver version
 RUN nvidia-smi
 
 # # Set DEBIAN_FRONTEND to noninteractive to prevent interactive prompts