adding pytorch build

WIPACrepo · Sep 19, 2024 · d0d715d · d0d715d
1 parent 98ef442
commit d0d715d
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/.github/workflows/build_ml.yml b/.github/workflows/build_ml.yml
@@ -110,7 +110,19 @@ jobs:
           tags: >
             icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-rocm5.7-ubuntu22.04-devel,
             icecube/icetray-ml:icetray-current-rocm5.7-ubuntu22.04-devel,
-            icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-rocm5.7-ubuntu22.04-devel-${{ env.current_date }}      
+            icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-rocm5.7-ubuntu22.04-devel-${{ env.current_date }}
+      -
+        name: Build and push icetray ml for 22 04 pytorch
+        uses: docker/build-push-action@v3
+        with:
+          push: true
+          file: icecube_ml/Dockerfile_ubuntu_22.04_cuda12.1_cudann_pytorch2.2.2
+          build-args: |
+            VERSION=${{ env.ICETRAY_VERSION }}
+          tags: >
+            icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda12.1-cudann-pytorch2.2.2-ubuntu22.04-devel,
+            icecube/icetray-ml:icetray-current-cuda12.1-cudann-pytorch2.2.2-ubuntu22.04-devel,
+            icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda12.1-cudann-pytorch2.2.2-ubuntu22.04-devel-${{ env.current_date }}
       # -
       #   name: Build and push icetray ml  20 04
       #   uses: docker/build-push-action@v3