Updated Dockerfile for GPU Dependencies

Dockerfile

@@ -6,11 +6,6 @@ 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