From ff1ec217073adbfc5dd9117c5315536aa4c6f555 Mon Sep 17 00:00:00 2001 From: Maryam Tahhan Date: Mon, 16 Sep 2024 11:21:39 -0400 Subject: [PATCH] fix(aa66ada): readding libraries to the builder The builder stage for the Kepler image needs to have the dcgm/habana libraries installed for the build tags to work. Signed-off-by: Maryam Tahhan --- build/Dockerfile | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/build/Dockerfile b/build/Dockerfile index cc6e1e97d7..7af21690e8 100644 --- a/build/Dockerfile +++ b/build/Dockerfile @@ -1,8 +1,21 @@ FROM quay.io/sustainable_computing_io/kepler_builder:ubi-9-libbpf-1.3.0 AS builder +ARG INSTALL_DCGM=false +ARG INSTALL_HABANA=false WORKDIR /workspace COPY . . +RUN if [[ "$INSTALL_DCGM" == "true" ]]; then \ + dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/cuda-rhel9.repo; \ + yum install -y datacenter-gpu-manager; \ +fi; + +RUN if [[ "$INSTALL_HABANA" == "true" ]]; then \ + rpm -Uvh https://vault.habana.ai/artifactory/rhel/9/9.2/habanalabs-firmware-tools-1.15.1-15.el9.x86_64.rpm --nodeps; \ + echo /usr/lib/habanalabs > /etc/ld.so.conf.d/habanalabs.conf; \ + ldconfig; \ +fi; + RUN make tidy-vendor format RUN make build