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: