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 && \