add pytorch 2.5.0 base images (#1979)

* add pytorch 2.5.0 base images * make sure num examples for debug is zero and fix comparison
2024-10-18 03:36:51 -04:00
parent f62e23737b
commit 67f744dc8c
3 changed files with 8 additions and 2 deletions
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -36,6 +36,12 @@ jobs:
            python_version: "3.11"
            pytorch: 2.4.1
            torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
+          - cuda: "124"
+            cuda_version: 12.4.1
+            cudnn_version: ""
+            python_version: "3.11"
+            pytorch: 2.5.0
+            torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
    steps:
      - name: Checkout
        uses: actions/checkout@v3