fix platform for GPU tests

allenai · Mar 14, 2022 · 7bcc749 · 7bcc749
1 parent eca99b7
commit 7bcc749
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -124,7 +124,7 @@ jobs:
         - name: GPU Tests
           runs_on: [self-hosted, GPU]
           coverage_report: true
-          torch_platform: cu111  # Our self-hosted GPU runners currently support CUDA 11.*
+          torch_platform: cu113  # Our self-hosted GPU runners currently support CUDA 11.*
           run: make gpu-tests
 
     steps: