Skip to content

Commit 3b37c3e

Browse files
authored
docker: update building base docker images for last CUDA & py3.10 (#20844)
docker: extend building base docker images for last CUDA
1 parent 29f0b13 commit 3b37c3e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/docker-build.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,8 @@ jobs:
9797
# adding dome more images as Thunder mainly using python 3.10,
9898
# and we need to support integrations as for example LitGPT
9999
python_version: ["3.10"]
100-
pytorch_version: ["2.3.1", "2.4.1", "2.5.1", "2.6.0", "2.7.0"]
101-
cuda_version: ["12.4.1"]
100+
pytorch_version: ["2.6.0", "2.7.0"]
101+
cuda_version: ["12.6.3"]
102102
include:
103103
# These are the base images for PL release docker images.
104104
# Make sure the matrix here matches the one above.

0 commit comments

Comments
 (0)