add missing dockerfile dependencies

Signed-off-by: Varun Ramachandra Sekar <[email protected]>

deployments/container/Dockerfile.ubi8

@@ -19,7 +19,7 @@ ARG BASE_DIST=ubi8
 FROM --platform=${TARGETARCH} nvidia/cuda:${CUDA_VERSION}-base-${BASE_DIST} as build
 
 RUN yum install -y \
-    wget make git gcc \
+    wget make git gcc kmod \
      && \
     rm -rf /var/cache/yum/*
 
@@ -63,6 +63,8 @@ COPY --from=build /artifacts/nvidia-dra-plugin         /usr/bin/nvidia-dra-plugi
 COPY --from=build /build/templates                     /templates
 COPY --from=build /build/scripts/bind_to_driver.sh     /usr/bin/bind_to_driver.sh
 COPY --from=build /build/scripts/unbind_from_driver.sh /usr/bin/unbind_from_driver.sh
+COPY --from=build /usr/bin/nsenter                     /usr/bin/nsenter
+COPY --from=build /usr/sbin/modprobe                   /usr/sbin/modprobe
 
 # Install / upgrade packages here that are required to resolve CVEs
 ARG CVE_UPDATES

deployments/container/Dockerfile.ubuntu

@@ -19,7 +19,7 @@ ARG BASE_DIST=ubuntu20.04
 FROM --platform=${BUILDOS}/amd64 nvidia/cuda:${CUDA_VERSION}-base-${BASE_DIST} as build
 
 RUN apt-get update && \
-    apt-get install -y wget make git gcc-aarch64-linux-gnu gcc \
+    apt-get install -y wget make git gcc-aarch64-linux-gnu gcc kmod \
     && \
     rm -rf /var/lib/apt/lists/*
 
@@ -68,6 +68,8 @@ COPY --from=build /artifacts/nvidia-dra-plugin         /usr/bin/nvidia-dra-plugi
 COPY --from=build /build/templates                     /templates
 COPY --from=build /build/scripts/bind_to_driver.sh     /usr/bin/bind_to_driver.sh
 COPY --from=build /build/scripts/unbind_from_driver.sh /usr/bin/unbind_from_driver.sh
+COPY --from=build /usr/bin/nsenter                     /usr/bin/nsenter
+COPY --from=build /usr/sbin/modprobe                   /usr/sbin/modprobe
 
 # Install / upgrade packages here that are required to resolve CVEs
 ARG CVE_UPDATES