43
43
# Using ubuntu-20.04 instead of 22.04 for more compatibility (glibc). Ideally we'd use the
44
44
# manylinux docker image, but I haven't figured out how to install CUDA on manylinux.
45
45
os : [ubuntu-20.04]
46
- python-version : ['3.7', '3. 8', '3.9', '3.10', '3.11', '3.12']
47
- torch-version : ['1.12.1', '1.13.1', ' 2.0.1', '2.1.2', '2.2.2', '2.3.0 ', '2.4.0.dev20240420 ']
46
+ python-version : ['3.8', '3.9', '3.10', '3.11', '3.12']
47
+ torch-version : ['2.0.1', '2.1.2', '2.2.2', '2.3.1 ', '2.4.0.dev20240505 ']
48
48
cuda-version : ['11.8.0', '12.2.2']
49
49
# We need separate wheels that either uses C++11 ABI (-D_GLIBCXX_USE_CXX11_ABI) or not.
50
50
# Pytorch wheels currently don't use it, but nvcr images have Pytorch compiled with C++11 ABI.
@@ -53,33 +53,11 @@ jobs:
53
53
cxx11_abi : ['FALSE', 'TRUE']
54
54
exclude :
55
55
# Pytorch < 2.2 does not support Python 3.12
56
- - torch-version : ' 1.12.1'
57
- python-version : ' 3.12'
58
- - torch-version : ' 1.13.1'
59
- python-version : ' 3.12'
60
56
- torch-version : ' 2.0.1'
61
57
python-version : ' 3.12'
62
58
- torch-version : ' 2.1.2'
63
59
python-version : ' 3.12'
64
- # Pytorch <= 1.12 does not support Python 3.11
65
- - torch-version : ' 1.12.1'
66
- python-version : ' 3.11'
67
- # Pytorch >= 2.0 only supports Python >= 3.8
68
- - torch-version : ' 2.0.1'
69
- python-version : ' 3.7'
70
- - torch-version : ' 2.1.2'
71
- python-version : ' 3.7'
72
- - torch-version : ' 2.2.2'
73
- python-version : ' 3.7'
74
- - torch-version : ' 2.3.0'
75
- python-version : ' 3.7'
76
- - torch-version : ' 2.4.0.dev20240420'
77
- python-version : ' 3.7'
78
60
# Pytorch <= 2.0 only supports CUDA <= 11.8
79
- - torch-version : ' 1.12.1'
80
- cuda-version : ' 12.2.2'
81
- - torch-version : ' 1.13.1'
82
- cuda-version : ' 12.2.2'
83
61
- torch-version : ' 2.0.1'
84
62
cuda-version : ' 12.2.2'
85
63
@@ -138,8 +116,8 @@ jobs:
138
116
# e.g. we can have system CUDA version being 11.7 but if torch==1.12 then we need to download the wheel from cu116
139
117
# This code is ugly, maybe there's a better way to do this.
140
118
export TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \
141
- minv = {'1.12': 113, '1.13': 116, ' 2.0': 117, '2.1': 118, '2.2': 118, '2.3': 118, '2.4': 118}[env['MATRIX_TORCH_VERSION']]; \
142
- maxv = {'1.12': 116, '1.13': 117, ' 2.0': 118, '2.1': 121, '2.2': 121, '2.3': 121, '2.4': 121}[env['MATRIX_TORCH_VERSION']]; \
119
+ minv = {'2.0': 117, '2.1': 118, '2.2': 118, '2.3': 118, '2.4': 118}[env['MATRIX_TORCH_VERSION']]; \
120
+ maxv = {'2.0': 118, '2.1': 121, '2.2': 121, '2.3': 121, '2.4': 121}[env['MATRIX_TORCH_VERSION']]; \
143
121
print(max(min(int(env['MATRIX_CUDA_VERSION']), maxv), minv))" \
144
122
)
145
123
if [[ ${{ matrix.torch-version }} == *"dev"* ]]; then
0 commit comments