update to latest nccl in docker image (#965)

axolotl-ai-cloud · Dec 16, 2023 · 80ec7af · 80ec7af
1 parent f28e755
commit 80ec7af
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
+    apt-get install -y vim curl nano libnccl2 libnccl-dev
 
 WORKDIR /workspace