Update Dockerfile

Dockerfile

@@ -3,16 +3,20 @@ FROM nvidia/cuda:12.2.0-devel-ubuntu20.04
 ENV DEBIAN_FRONTEND=noninteractive
 
 # Add NVIDIA CUDA Repository
-RUN apt update && apt install -y wget gnupg software-properties-common && \
+RUN apt update && apt install -y wget gnupg && \
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \
     mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
     apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \
     add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
 
-# Install CUDA components separately
+# Install Specific libcublas Version
 RUN apt update && \
-    apt install -y cuda-libraries-12-2 cuda-libraries-dev-12-2 && \
-    apt install --no-install-recommends -y build-essential python3 python3-pip curl git cmake zlib1g-dev libblas-dev && \
+    apt install -y libcublas-12-2=12.2.5.6 libcublas-dev-12-2=12.2.5.6 && \
+    apt clean && rm -rf /var/lib/apt/lists/*
+
+# Install Remaining Packages
+RUN apt update && \
+    apt install --no-install-recommends -y cuda-libraries-12-2 cuda-libraries-dev-12-2 build-essential python3 python3-pip curl git cmake zlib1g-dev libblas-dev && \
     apt clean && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app