Skip to content

Commit c9b1ffe

Browse files
committed
docker: extend building base docker images for litGPT
1 parent 3d39824 commit c9b1ffe

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)