modify Dockerfile to stop pulling cuda:12.9.1-ubi9 base image

Signed-off-by: Tariq Ibrahim <[email protected]>

Author: tariq1890

docker/Dockerfile

@@ -40,19 +40,18 @@ RUN make cmds
 
 FROM nvcr.io/nvidia/cuda:13.0.1-base-ubi9 AS cuda-base
 
-RUN dnf install -y --allowerasing \
-        curl \
-    && \
-        dnf clean all
+FROM registry.access.redhat.com/ubi9/ubi:latest AS sample-builder
 
-WORKDIR /workspace
+ARG CUDA_COMPAT_PKG_VERSION=12-9
 
-FROM nvcr.io/nvidia/cuda:12.9.1-base-ubi9 AS sample-builder
+RUN ARCH=$(uname -m) && OS_ARCH=${ARCH/amd64/x86_64} && OS_ARCH=${ARCH/aarch64/sbsa} && OS_ARCH=${ARCH/arm64/sbsa} \
+    dnf config-manager addrepo --from-repofile=https://developer.download.nvidia.com/compute/cuda/repos/rhel9/$OS_ARCH/cuda-rhel9.repo
 
 RUN dnf install -y --allowerasing \
         cmake \
-        cuda-cudart-devel-12-9 \
-        cuda-nvcc-12-9 \
+        cuda-cudart-devel-${CUDA_COMPAT_PKG_VERSION} \
+        cuda-compat-${CUDA_COMPAT_PKG_VERSION} \
+        cuda-nvcc-${CUDA_COMPAT_PKG_VERSION} \
         g++ \
         curl \
     && \