diff --git a/build/container/Dockerfile.centos b/build/container/Dockerfile.centos index 26b6bf95..c6a5b46b 100644 --- a/build/container/Dockerfile.centos +++ b/build/container/Dockerfile.centos @@ -75,4 +75,11 @@ LABEL description="See summary" COPY ./LICENSE /licenses/LICENSE -ENTRYPOINT ["/work/nvidia-toolkit"] \ No newline at end of file +# Install / upgrade packages here that are required to resolve CVEs +ARG CVE_UPDATES +RUN if [ -n "${CVE_UPDATES}" ]; then \ + yum update -y ${CVE_UPDATES} && \ + rm -rf /var/cache/yum/*; \ + fi + +ENTRYPOINT ["/work/nvidia-toolkit"] diff --git a/build/container/Makefile b/build/container/Makefile index c62e5b58..100c22c1 100644 --- a/build/container/Makefile +++ b/build/container/Makefile @@ -75,6 +75,7 @@ $(BUILD_TARGETS): build-%: $(ARTIFACTS_ROOT) --build-arg PACKAGE_VERSION="$(PACKAGE_VERSION)" \ --build-arg PACKAGE_ARCH="$(PACKAGE_ARCH)" \ --build-arg VERSION="$(VERSION)" \ + --build-arg CVE_UPDATES="$(CVE_UPDATES)" \ -f $(DOCKERFILE) \ $(CURDIR)