Skip to content

Commit 747bb19

Browse files
committed
debug: missing nvcc on images with pt-2.1
1 parent 634e6e6 commit 747bb19

File tree

3 files changed

+11
-5
lines changed

3 files changed

+11
-5
lines changed

.lightning/workflows/fabric.yml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ parametrize:
1212
# note that this is setting also all oldest requirements which is linked to Torch == 2.0
1313
- image: "pytorchlightning/pytorch_lightning:base-cuda12.1.1-py3.10-torch2.1"
1414
PACKAGE_NAME: "fabric"
15-
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"
15+
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.8"
1616
PACKAGE_NAME: "fabric"
1717
# - image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"
1818
# PACKAGE_NAME: "fabric"
19-
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"
19+
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.8"
2020
PACKAGE_NAME: "lightning"
2121
exclude: []
2222

@@ -27,6 +27,9 @@ env:
2727
run: |
2828
whereis nvidia
2929
nvidia-smi
30+
ls -l /usr/local/cuda/bin/nvcc
31+
which nvcc
32+
nvcc --version
3033
python --version
3134
pip --version
3235
pip install -q fire wget packaging

.lightning/workflows/pytorch.yml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ parametrize:
1212
# note that this is setting also all oldest requirements which is linked to Torch == 2.0
1313
- image: "pytorchlightning/pytorch_lightning:base-cuda12.1.1-py3.10-torch2.1"
1414
PACKAGE_NAME: "pytorch"
15-
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"
15+
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.8"
1616
PACKAGE_NAME: "pytorch"
1717
# - image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"
1818
# PACKAGE_NAME: "pytorch"
19-
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"
19+
- image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.8"
2020
PACKAGE_NAME: "lightning"
2121
exclude: []
2222

@@ -27,6 +27,9 @@ env:
2727
run: |
2828
whereis nvidia
2929
nvidia-smi
30+
ls -l /usr/local/cuda/bin/nvcc
31+
which nvcc
32+
nvcc --version
3033
python --version
3134
pip --version
3235
pip install -q fire wget packaging

dockers/base-cuda/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# limitations under the License.
1414

1515
ARG UBUNTU_VERSION=22.04
16-
ARG CUDA_VERSION=11.7.1
16+
ARG CUDA_VERSION=12.1.1
1717

1818

1919
FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}

0 commit comments

Comments
 (0)