configure nightly docker builds (#1454) [skip ci]

* configure nightly docker builds * also test update pytorch in modal ci
2024-03-29 08:25:45 -04:00
parent 05b398a072
commit 8cb127abeb
3 changed files with 122 additions and 5 deletions
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -79,6 +79,11 @@ jobs:
            python_version: "3.10"
            pytorch: 2.1.2
            num_gpus: 1
+          - cuda: 121
+            cuda_version: 12.1.0
+            python_version: "3.11"
+            pytorch: 2.2.1
+            num_gpus: 1
    steps:
      - name: Checkout
        uses: actions/checkout@v4