refine

Oneflow-Inc · Dec 20, 2023 · 5ea97ee · 5ea97ee
1 parent ccef749
commit 5ea97ee
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/test-with-pytorch.yml b/.github/workflows/test-with-pytorch.yml
@@ -31,6 +31,7 @@ jobs:
             1.9.0-cuda10.2-cudnn7-runtime,
             1.10.0-cuda11.3-cudnn8-runtime,
             1.13.1-cuda11.6-cudnn8-runtime,
+            2.1.2-cuda12.1-cudnn8-runtime,
           ]
     env:
       DOCKER_REPO: test-with-pytorch-${{matrix.pytorch-tag}}