Skip to content

Commit b5931e9

Browse files
authored
[tensforlow] Release TF2.3 cuda110 training cpu and gpu (#1078)
1 parent 6022443 commit b5931e9

File tree

1 file changed

+33
-43
lines changed

1 file changed

+33
-43
lines changed

release_images.yml

Lines changed: 33 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -53,23 +53,39 @@ release_images:
5353
device_types: ["cpu"]
5454
python_versions: ["py37"]
5555
os_version: "ubuntu18.04"
56-
cuda_version: "cu102"
56+
cuda_version: "cu110"
5757
example: False # [Default: False] Set to True to denote that this image is an Example image
5858
disable_sm_tag: False # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
5959
# to images being published.
6060
force_release: False # [Default: False] Set to True to force images to be published even if the same image
6161
# has already been published. Re-released image will have minor version incremented by 1.
62-
inference:
63-
device_types: ["cpu", "gpu"]
62+
5:
63+
framework: "tensorflow"
64+
version: "2.3.2"
65+
training:
66+
device_types: ["gpu"]
6467
python_versions: ["py37"]
6568
os_version: "ubuntu18.04"
66-
cuda_version: "cu102"
69+
cuda_version: "cu110"
6770
example: False # [Default: False] Set to True to denote that this image is an Example image
71+
disable_sm_tag: True # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
72+
# to images being published.
73+
force_release: False # [Default: False] Set to True to force images to be published even if the same image
74+
# has already been published. Re-released image will have minor version incremented by 1.
75+
6:
76+
framework: "tensorflow"
77+
version: "2.3.2"
78+
training:
79+
device_types: ["gpu"]
80+
python_versions: ["py37"]
81+
os_version: "ubuntu18.04"
82+
cuda_version: "cu110"
83+
example: True # [Default: False] Set to True to denote that this image is an Example image
6884
disable_sm_tag: False # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
6985
# to images being published.
7086
force_release: False # [Default: False] Set to True to force images to be published even if the same image
7187
# has already been published. Re-released image will have minor version incremented by 1.
72-
5:
88+
7:
7389
framework: "tensorflow"
7490
version: "1.15.5"
7591
training:
@@ -88,7 +104,7 @@ release_images:
88104
example: False
89105
disable_sm_tag: False
90106
force_release: False
91-
6:
107+
8:
92108
framework: "tensorflow"
93109
version: "1.15.5"
94110
training:
@@ -101,7 +117,7 @@ release_images:
101117
# to images being published.
102118
force_release: False # [Default: False] Set to True to force images to be published even if the same image
103119
# has already been published. Re-released image will have minor version incremented by 1.
104-
7:
120+
9:
105121
framework: "tensorflow"
106122
version: "2.3.0"
107123
inference:
@@ -113,7 +129,7 @@ release_images:
113129
# to images being published.
114130
force_release: False # [Default: False] Set to True to force images to be published even if the same image
115131
# has already been published. Re-released image will have minor version incremented by 1.
116-
8:
132+
10:
117133
framework: "mxnet"
118134
version: "1.8.0"
119135
training:
@@ -134,7 +150,7 @@ release_images:
134150
example: False
135151
disable_sm_tag: False
136152
force_release: False
137-
9:
153+
11:
138154
framework: "mxnet"
139155
version: "1.8.0"
140156
training:
@@ -145,7 +161,7 @@ release_images:
145161
example: True
146162
disable_sm_tag: False # [Default: False] This option is not used by Example images
147163
force_release: False
148-
10:
164+
12:
149165
framework: "mxnet"
150166
version: "1.7.0"
151167
training:
@@ -164,7 +180,7 @@ release_images:
164180
example: False
165181
disable_sm_tag: False # [Default: False] This option is not used by Example images
166182
force_release: False
167-
11:
183+
13:
168184
framework: "mxnet"
169185
version: "1.7.0"
170186
training:
@@ -175,7 +191,7 @@ release_images:
175191
example: True
176192
disable_sm_tag: False # [Default: False] This option is not used by Example images
177193
force_release: False
178-
12:
194+
14:
179195
framework: "tensorflow"
180196
version: "2.4.1"
181197
training:
@@ -194,7 +210,7 @@ release_images:
194210
example: False
195211
disable_sm_tag: False
196212
force_release: False
197-
13:
213+
15:
198214
framework: "tensorflow"
199215
version: "2.4.1"
200216
training:
@@ -205,7 +221,7 @@ release_images:
205221
example: True
206222
disable_sm_tag: False
207223
force_release: False
208-
14:
224+
16:
209225
framework: "pytorch"
210226
version: "1.8.1"
211227
training:
@@ -228,7 +244,7 @@ release_images:
228244
# to images being published.
229245
force_release: False # [Default: False] Set to True to force images to be published even if the same image
230246
# has already been published. Re-released image will have minor version incremented by 1.
231-
15:
247+
17:
232248
framework: "pytorch"
233249
version: "1.8.1"
234250
training:
@@ -239,7 +255,7 @@ release_images:
239255
example: True
240256
disable_sm_tag: False # [Default: False] This option is not used by Example images
241257
force_release: False
242-
16:
258+
18:
243259
framework: "pytorch"
244260
version: "1.5.1"
245261
training:
@@ -258,7 +274,7 @@ release_images:
258274
example: False
259275
disable_sm_tag: False # [Default: False] This option is not used by Example images
260276
force_release: False
261-
17:
277+
19:
262278
framework: "pytorch"
263279
version: "1.5.1"
264280
training:
@@ -269,32 +285,6 @@ release_images:
269285
example: True
270286
disable_sm_tag: False # [Default: False] This option is not used by Example images
271287
force_release: False
272-
18:
273-
framework: "tensorflow"
274-
version: "2.3.2"
275-
training:
276-
device_types: ["gpu"]
277-
python_versions: ["py37"]
278-
os_version: "ubuntu18.04"
279-
cuda_version: "cu110"
280-
example: False # [Default: False] Set to True to denote that this image is an Example image
281-
disable_sm_tag: True # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
282-
# to images being published.
283-
force_release: False # [Default: False] Set to True to force images to be published even if the same image
284-
# has already been published. Re-released image will have minor version incremented by 1.
285-
19:
286-
framework: "tensorflow"
287-
version: "2.3.2"
288-
training:
289-
device_types: ["gpu"]
290-
python_versions: ["py37"]
291-
os_version: "ubuntu18.04"
292-
cuda_version: "cu110"
293-
example: True # [Default: False] Set to True to denote that this image is an Example image
294-
disable_sm_tag: False # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
295-
# to images being published.
296-
force_release: False # [Default: False] Set to True to force images to be published even if the same image
297-
# has already been published. Re-released image will have minor version incremented by 1.
298288
20:
299289
framework: "huggingface_tensorflow"
300290
version: "2.4.1"

0 commit comments

Comments
 (0)