From 20b8de4f31a348a379778b0012305d68deb4a7f5 Mon Sep 17 00:00:00 2001
From: Liam Huang <4506671+llama1969@users.noreply.github.com>
Date: Fri, 20 Sep 2024 20:02:55 +0800
Subject: [PATCH] Fix building Docker images for AWS CSP tools.

---
 csp_tools/aws/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/csp_tools/aws/Dockerfile b/csp_tools/aws/Dockerfile
index 12c0cb7a9..52c687a40 100755
--- a/csp_tools/aws/Dockerfile
+++ b/csp_tools/aws/Dockerfile
@@ -23,7 +23,7 @@ RUN cd /opt && \
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb && \
     dpkg -i cuda-keyring_1.0-1_all.deb && \
     apt update && \
-    apt install libnccl2==${NCCL_VERSION} libnccl-dev==${NCCL_VERSION}
+    apt install -y --allow-downgrades libnccl2=${NCCL_VERSION} libnccl-dev=${NCCL_VERSION}
 
 # EFA
 RUN apt-get update && \