Skip to content

Commit a8f2c4d

Browse files
Upgrade NCCL to 2.26.2 in Kubeflow CUDA training image
1 parent 05559c7 commit a8f2c4d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

images/runtime/training/cuda/Dockerfile

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,9 @@ RUN micropipenv install -- --no-cache-dir && \
8383
RUN pip install wheel
8484
RUN pip install --no-cache-dir flash-attn==2.7.4.post1 --no-build-isolation
8585

86+
# Upgrade NCCL to a more recent version until we upgrade torch
87+
RUN pip install nvidia-nccl-cu12==2.26.2
88+
8689
# Restore user workspace
8790
USER 1001
8891

0 commit comments

Comments
 (0)