Skip to content

Commit c37238e

Browse files
committed
undo cuda pinning
1 parent cf807f3 commit c37238e

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

.lightning/workflows/fabric.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@ run: |
4949
libopenmpi-dev \
5050
openmpi-bin \
5151
ninja-build \
52+
libnccl2 \
53+
libnccl-dev
5254
5355
echo "Install Python ${python_version} and UV"
5456
apt-get install -y python${python_version} python${python_version}-venv python${python_version}-dev

.lightning/workflows/pytorch.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,8 @@ run: |
4848
libopenmpi-dev \
4949
openmpi-bin \
5050
ninja-build \
51+
libnccl2 \
52+
libnccl-dev \
5153
unzip
5254
5355
echo "Install Python ${python_version} and UV"

0 commit comments

Comments
 (0)