Skip to content

Commit bef53a8

Browse files
authored
docker: extend building base docker images for litGPT (#20842)
1 parent 760ae74 commit bef53a8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

.github/workflows/docker-build.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,11 @@ jobs:
9494
strategy:
9595
fail-fast: false
9696
matrix:
97+
# adding dome more images as Thunder mainly using python 3.10,
98+
# and we need to support integrations as for example LitGPT
99+
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"]
97102
include:
98103
# These are the base images for PL release docker images.
99104
# Make sure the matrix here matches the one above.

0 commit comments

Comments
 (0)