Skip to content

Commit 70618bc

Browse files
committed
pin cuda version
1 parent 5d6d95e commit 70618bc

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

.lightning/workflows/fabric.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ run: |
4949
libopenmpi-dev \
5050
openmpi-bin \
5151
ninja-build \
52-
libnccl2 \
52+
libnccl2=2.28.7 \
5353
libnccl-dev=2.28.7
5454
5555
echo "Install Python ${python_version} and UV"

.lightning/workflows/pytorch.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ run: |
4848
libopenmpi-dev \
4949
openmpi-bin \
5050
ninja-build \
51-
libnccl2 \
51+
libnccl2=2.28.7 \
5252
libnccl-dev=2.28.7 \
5353
unzip
5454

0 commit comments

Comments
 (0)