Skip to content

Commit 7f23d3d

Browse files
committed
[release] fix cuda versions
1 parent 9f18a5d commit 7f23d3d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/wheels.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ jobs:
1313
strategy:
1414
matrix:
1515
python_version: [["3.7", "cp37-cp37m"], ["3.8", "cp38-cp38"], ["3.9", "cp39-cp39"], ["3.10", "cp310-cp310"]]
16-
cuda_support: [["", "--extra-index-url https://download.pytorch.org/whl/cpu", "\"['cpu', 'cu113', 'cu116']\"", "cpu"], ["+cu102", "", "\"['cu102']\"", "cuda102"]]
16+
cuda_support: [["", "--extra-index-url https://download.pytorch.org/whl/cpu", "\"['cpu', '11.3', '11.6']\"", "cpu"], ["+cu102", "", "\"['10.2']\"", "cuda102"]]
1717
container: pytorch/manylinux-${{ matrix.cuda_support[3] }}
1818
steps:
1919
- name: Checkout functorch

0 commit comments

Comments
 (0)