From 7f23d3dd67ebbb02acc268c3381fd5815a395808 Mon Sep 17 00:00:00 2001 From: Richard Zou Date: Thu, 30 Jun 2022 15:24:19 -0700 Subject: [PATCH] [release] fix cuda versions --- .github/workflows/wheels.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml index 486c12673..2ba0fe1e3 100644 --- a/.github/workflows/wheels.yml +++ b/.github/workflows/wheels.yml @@ -13,7 +13,7 @@ jobs: strategy: matrix: python_version: [["3.7", "cp37-cp37m"], ["3.8", "cp38-cp38"], ["3.9", "cp39-cp39"], ["3.10", "cp310-cp310"]] - cuda_support: [["", "--extra-index-url https://download.pytorch.org/whl/cpu", "\"['cpu', 'cu113', 'cu116']\"", "cpu"], ["+cu102", "", "\"['cu102']\"", "cuda102"]] + cuda_support: [["", "--extra-index-url https://download.pytorch.org/whl/cpu", "\"['cpu', '11.3', '11.6']\"", "cpu"], ["+cu102", "", "\"['10.2']\"", "cuda102"]] container: pytorch/manylinux-${{ matrix.cuda_support[3] }} steps: - name: Checkout functorch