Merge pull request #198 from red-hat-data-services/revert-196-test3

Revert "updating docker file"

Author: m-rafeeq

.tekton/distributed-workloads-poc1-push.yaml

@@ -74,7 +74,7 @@ spec:
     - default: ""
       description: Build dependencies to be prefetched
       name: prefetch-input
-      value: '[{"type": "pipenv", "path": "images/runtime/training/py312-cuda128-torch280"}, {"type": "rpm", "path": "images/runtime/training/py312-cuda128-torch280"}]'
+      value: '{"type": "pipenv", "path": "images/runtime/training/py312-cuda128-torch280"}'
       type: string
     - default: ""
       description: Image tag expiration time, time values could be something like

images/runtime/training/py312-cuda128-torch280/Dockerfile

@@ -34,8 +34,11 @@ ENV CUDA_VERSION=12.8.0 \
 
 # Ref: https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.8.0/ubi9/base/Dockerfile
 # nvcc is required for Flash Attention
-# CUDA packages are now managed via rpms.lock.yaml - install from locked versions
-RUN dnf install -y \
+RUN dnf config-manager \
+    --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/cuda-rhel9.repo \
+ && dnf install -y \
+     --disablerepo=rhel-9-for-x86_64-baseos-rpms \
+     --disablerepo=rhel-9-for-x86_64-appstream-rpms \
      cuda-cudart-12-8-${NV_CUDA_CUDART_VERSION} \
      cuda-compat-12-8-${NV_CUDA_COMPAT_VERSION} \
      cuda-nvcc-12-8-${NV_CUDA_NVCC_VERSION} \
@@ -48,8 +51,10 @@ ENV CUDA_HOME="/usr/local/cuda" \
  LD_LIBRARY_PATH="/usr/local/nvidia/lib:/usr/local/nvidia/lib64:$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64:$LD_LIBRARY_PATH"
 
 # Install InfiniBand and RDMA packages
-# InfiniBand/RDMA packages are now managed via rpms.lock.yaml - install from locked versions
-RUN dnf install -y \
+RUN dnf config-manager \
+        --add-repo https://linux.mellanox.com/public/repo/mlnx_ofed/latest/rhel9.5/mellanox_mlnx_ofed.repo 
+
+RUN dnf install -y --disablerepo="*" --enablerepo="cuda-rhel9-x86_64,mlnx_ofed_24.10-1.1.4.0_base,ubi-9-appstream-rpms,ubi-9-baseos-rpms" \
         libibverbs-utils \
         infiniband-diags \
         libibumad3 \