Skip to content

Commit 615ad96

Browse files
caponettocaponetto
committed
Add Tensorflow/PyTorch version compatibility
1 parent e506603 commit 615ad96

File tree

2 files changed

+780
-0
lines changed

2 files changed

+780
-0
lines changed

‎ci/version-compatibility/pytorch.json

Lines changed: 184 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,184 @@
1+
{
2+
"2.3.1": {
3+
"python": {
4+
"min": "3.8",
5+
"max": "3.11"
6+
},
7+
"torchvision": "0.18.1",
8+
"torchaudio": "2.3.0",
9+
"gpu": {
10+
"cuda": "11.8",
11+
"cudnn": "8.7.0.84",
12+
"rocm": "6.0"
13+
}
14+
},
15+
"2.3.0": {
16+
"python": {
17+
"min": "3.8",
18+
"max": "3.11"
19+
},
20+
"torchvision": "0.18.0",
21+
"torchaudio": "2.3.0",
22+
"gpu": {
23+
"cuda": "11.8",
24+
"cudnn": "8.7.0.84",
25+
"rocm": "6.0"
26+
}
27+
},
28+
"2.2.2": {
29+
"python": {
30+
"min": "3.8",
31+
"max": "3.11"
32+
},
33+
"torchvision": "0.17.2",
34+
"torchaudio": "2.2.2",
35+
"gpu": {
36+
"cuda": "11.8",
37+
"cudnn": "8.7.0.84",
38+
"rocm": "5.7"
39+
}
40+
},
41+
"2.2.1": {
42+
"python": {
43+
"min": "3.8",
44+
"max": "3.11"
45+
},
46+
"torchvision": "0.17.1",
47+
"torchaudio": "2.2.1",
48+
"gpu": {
49+
"cuda": "11.8",
50+
"cudnn": "8.7.0.84",
51+
"rocm": "5.7"
52+
}
53+
},
54+
"2.2.0": {
55+
"python": {
56+
"min": "3.8",
57+
"max": "3.11"
58+
},
59+
"torchvision": "0.17.0",
60+
"torchaudio": "2.2.0",
61+
"gpu": {
62+
"cuda": "11.8",
63+
"cudnn": "8.7.0.84",
64+
"rocm": "5.7"
65+
}
66+
},
67+
"2.1.2": {
68+
"python": {
69+
"min": "3.8",
70+
"max": "3.11"
71+
},
72+
"torchvision": "0.16.2",
73+
"torchaudio": "2.1.2",
74+
"gpu": {
75+
"cuda": "11.8",
76+
"cudnn": "8.7.0.84",
77+
"rocm": "5.6"
78+
}
79+
},
80+
"2.1.1": {
81+
"python": {
82+
"min": "3.8",
83+
"max": "3.11"
84+
},
85+
"torchvision": "0.16.1",
86+
"torchaudio": "2.1.1",
87+
"gpu": {
88+
"cuda": "11.8",
89+
"cudnn": "8.7.0.84",
90+
"rocm": "5.6"
91+
}
92+
},
93+
"2.1.0": {
94+
"python": {
95+
"min": "3.8",
96+
"max": "3.11"
97+
},
98+
"torchvision": "0.16.0",
99+
"torchaudio": "2.1.0",
100+
"gpu": {
101+
"cuda": "11.8",
102+
"cudnn": "8.7.0.84",
103+
"rocm": "5.6"
104+
}
105+
},
106+
"2.0.1": {
107+
"python": {
108+
"min": "3.8",
109+
"max": "3.11"
110+
},
111+
"torchvision": "0.15.2",
112+
"torchaudio": "2.0.2",
113+
"gpu": {
114+
"cuda": "11.7",
115+
"cudnn": "8.5.0.96",
116+
"rocm": "5.4"
117+
}
118+
},
119+
"2.0.0": {
120+
"python": {
121+
"min": "3.8",
122+
"max": "3.11"
123+
},
124+
"torchvision": "0.15.0",
125+
"torchaudio": "2.0.0",
126+
"gpu": {
127+
"cuda": "11.7",
128+
"cudnn": "8.5.0.96",
129+
"rocm": "5.4"
130+
}
131+
},
132+
"1.13.1": {
133+
"python": {
134+
"min": "3.7",
135+
"max": "3.10"
136+
},
137+
"torchvision": "0.14.1",
138+
"torchaudio": "0.13.1",
139+
"gpu": {
140+
"cuda": "11.6",
141+
"cudnn": "8.3.2.44",
142+
"rocm": "5.2"
143+
}
144+
},
145+
"1.13.0": {
146+
"python": {
147+
"min": "3.7",
148+
"max": "3.10"
149+
},
150+
"torchvision": "0.14.0",
151+
"torchaudio": "0.13.0",
152+
"gpu": {
153+
"cuda": "11.6",
154+
"cudnn": "8.3.2.44",
155+
"rocm": "5.2"
156+
}
157+
},
158+
"1.12.1": {
159+
"python": {
160+
"min": "3.7",
161+
"max": "3.10"
162+
},
163+
"torchvision": "0.13.1",
164+
"torchaudio": "0.12.1",
165+
"gpu": {
166+
"cuda": "11.3",
167+
"cudnn": "8.3.2.44",
168+
"rocm": "5.0"
169+
}
170+
},
171+
"1.12.0": {
172+
"python": {
173+
"min": "3.7",
174+
"max": "3.10"
175+
},
176+
"torchvision": "0.13.0",
177+
"torchaudio": "0.12.0",
178+
"gpu": {
179+
"cuda": "11.3",
180+
"cudnn": "8.3.2.44",
181+
"rocm": "5.0"
182+
}
183+
}
184+
}

0 commit comments

Comments
 (0)