From e944da0102fcf0f72d01d4006ebb4326f7408cbd Mon Sep 17 00:00:00 2001
From: iankouls-aws <83262950+iankouls-aws@users.noreply.github.com>
Date: Thu, 20 Jun 2024 17:36:24 -0700
Subject: [PATCH] Upgrade packages in container to reduce the number of CVEs
 (#361)

---
 micro-benchmarks/nccl-tests/nccl-tests.Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/micro-benchmarks/nccl-tests/nccl-tests.Dockerfile b/micro-benchmarks/nccl-tests/nccl-tests.Dockerfile
index 8bf7b7fe..ca221b26 100644
--- a/micro-benchmarks/nccl-tests/nccl-tests.Dockerfile
+++ b/micro-benchmarks/nccl-tests/nccl-tests.Dockerfile
@@ -8,7 +8,7 @@ ARG AWS_OFI_NCCL_VERSION=v1.8.1-aws
 ARG NCCL_VERSION=v2.20.3-1
 ARG NCCL_TESTS_VERSION=v2.13.9
 
-RUN apt-get update -y
+RUN apt-get update -y && apt-get upgrade -y
 RUN apt-get remove -y --allow-change-held-packages \
     ibverbs-utils \
     libibverbs-dev \