File tree Expand file tree Collapse file tree 3 files changed +11
-5
lines changed
Expand file tree Collapse file tree 3 files changed +11
-5
lines changed Original file line number Diff line number Diff 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
2727run : |
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
Original file line number Diff line number Diff 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
2727run : |
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
Original file line number Diff line number Diff line change 1313# limitations under the License.
1414
1515ARG UBUNTU_VERSION=22.04
16- ARG CUDA_VERSION=11.7 .1
16+ ARG CUDA_VERSION=12.1 .1
1717
1818
1919FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
You can’t perform that action at this time.
0 commit comments