We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 05559c7 commit a8f2c4dCopy full SHA for a8f2c4d
images/runtime/training/cuda/Dockerfile
@@ -83,6 +83,9 @@ RUN micropipenv install -- --no-cache-dir && \
83
RUN pip install wheel
84
RUN pip install --no-cache-dir flash-attn==2.7.4.post1 --no-build-isolation
85
86
+# Upgrade NCCL to a more recent version until we upgrade torch
87
+RUN pip install nvidia-nccl-cu12==2.26.2
88
+
89
# Restore user workspace
90
USER 1001
91
0 commit comments