fix for build for nccl in dockerfile (#970)

axolotl-ai-cloud · Dec 17, 2023 · 85de004 · 85de004
1 parent 80ec7af
commit 85de004
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -10,7 +10,7 @@ ARG PYTORCH_VERSION="2.0.1"
 ENV PYTORCH_VERSION=$PYTORCH_VERSION
 
 RUN apt-get update && \
-    apt-get install -y vim curl nano libnccl2 libnccl-dev
+    apt-get install -y --allow-change-held-packages vim curl nano libnccl2 libnccl-dev
 
 WORKDIR /workspace