[release] fix cuda versions

pytorch · Jun 30, 2022 · 7f23d3d · 7f23d3d
1 parent 9f18a5d
commit 7f23d3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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