push docker

Author: Borda

.github/workflows/docker-build.yml

@@ -28,7 +28,7 @@ concurrency:
   cancel-in-progress: ${{ github.event_name == 'pull_request' }}
 
 env:
-  PUSH_NIGHTLY: ${{ github.event_name == 'schedule' || github.event_name == 'workflow_dispatch' }}
+  PUSH_NIGHTLY: true # fixme: ${{ github.event_name == 'schedule' || github.event_name == 'workflow_dispatch' }}
   PUSH_RELEASE: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'release' }}
 
 jobs:
@@ -43,11 +43,11 @@ jobs:
         include:
           # We only release one docker image per PyTorch version.
           # Make sure the matrix here matches the one below.
-          - { python_version: "3.10", pytorch_version: "2.1", cuda_version: "12.1.0" }
-          - { python_version: "3.11", pytorch_version: "2.2", cuda_version: "12.1.0" }
-          - { python_version: "3.11", pytorch_version: "2.3", cuda_version: "12.1.0" }
-          - { python_version: "3.11", pytorch_version: "2.4", cuda_version: "12.1.0" }
-          - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.0" }
+          - { python_version: "3.10", pytorch_version: "2.1.0", cuda_version: "12.1.0" }
+          - { python_version: "3.11", pytorch_version: "2.2.2", cuda_version: "12.1.0" }
+          - { python_version: "3.11", pytorch_version: "2.3.1", cuda_version: "12.1.0" }
+          - { python_version: "3.11", pytorch_version: "2.4.1", cuda_version: "12.1.0" }
+          - { python_version: "3.12", pytorch_version: "2.5.0", cuda_version: "12.1.0" }
     steps:
       - uses: actions/checkout@v4
         with:
@@ -104,11 +104,11 @@ jobs:
         include:
           # These are the base images for PL release docker images.
           # Make sure the matrix here matches the one above.
-          - { python_version: "3.10", pytorch_version: "2.1", cuda_version: "12.1.0" }
-          - { python_version: "3.11", pytorch_version: "2.2", cuda_version: "12.1.0" }
-          - { python_version: "3.11", pytorch_version: "2.3", cuda_version: "12.1.0" }
-          - { python_version: "3.11", pytorch_version: "2.4", cuda_version: "12.1.0" }
-          - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.0" }
+          - { python_version: "3.10", pytorch_version: "2.1.0", cuda_version: "12.1.0" }
+          - { python_version: "3.11", pytorch_version: "2.2.2", cuda_version: "12.1.0" }
+          - { python_version: "3.11", pytorch_version: "2.3.1", cuda_version: "12.1.0" }
+          - { python_version: "3.11", pytorch_version: "2.4.1", cuda_version: "12.1.0" }
+          - { python_version: "3.12", pytorch_version: "2.5.0", cuda_version: "12.1.0" }
     steps:
       - uses: actions/checkout@v4
       - uses: docker/setup-buildx-action@v3
@@ -121,7 +121,7 @@ jobs:
       - name: shorten Torch version
         run: |
           # convert 1.10.2 to 1.10
-          pt_version=$(echo ${{ matrix.pytorch }} | cut -d. -f1,2)
+          pt_version=$(echo ${{ matrix.pytorch_version }} | cut -d. -f1,2)
           echo "PT_VERSION=$pt_version" >> $GITHUB_ENV
       - uses: docker/build-push-action@v6
         with:
@@ -131,7 +131,7 @@ jobs:
             CUDA_VERSION=${{ matrix.cuda_version }}
           file: dockers/base-cuda/Dockerfile
           push: ${{ env.PUSH_NIGHTLY }}
-          tags: "pytorchlightning/pytorch_lightning:base-cuda-py${{ matrix.python_version }}-torch${{ matrix.pytorch_version }}-cuda${{ matrix.cuda_version }}"
+          tags: "pytorchlightning/pytorch_lightning:base-cuda-py${{ matrix.python_version }}-torch${{ env.PT_VERSION }}-cuda${{ matrix.cuda_version }}"
         timeout-minutes: 95
       - uses: ravsamhq/notify-slack-action@v2
         if: failure() && env.PUSH_NIGHTLY == 'true'