From 7bcc74969ae804b5dca358145a00c4bbe7f360cf Mon Sep 17 00:00:00 2001 From: epwalsh Date: Mon, 14 Mar 2022 13:41:00 -0700 Subject: [PATCH] fix platform for GPU tests --- .github/workflows/ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index f4026dbda..f57bb70c9 100644 --- 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: