update to latest nccl in docker image (#965)

2023-12-16 18:31:25 -05:00
parent f28e75513b
commit 80ec7af358
1 changed files with 1 additions and 1 deletions
--- 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