Skip to content

[tensforlow] Release TF2.3 cuda110 training cpu and gpu #1078

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 7 commits into from
Apr 29, 2021
Merged

[tensforlow] Release TF2.3 cuda110 training cpu and gpu #1078

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
1796e98
release tf2.3 training cpu and gpu
jeet4320 Apr 29, 2021
0377e9b
release tf2.3 training cpu and gpu
jeet4320 Apr 29, 2021
2ea3365
add example
jeet4320 Apr 29, 2021
4885056
add example
jeet4320 Apr 29, 2021
854b6cb
more changes
jeet4320 Apr 29, 2021
37b60f6
more changes
jeet4320 Apr 29, 2021
41c288a
Merge branch 'master' into release-tf2.3-training
jeet4320 Apr 29, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
76 changes: 33 additions & 43 deletions release_images.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,23 +53,39 @@ release_images:
device_types: ["cpu"]
python_versions: ["py37"]
os_version: "ubuntu18.04"
cuda_version: "cu102"
cuda_version: "cu110"
example: False # [Default: False] Set to True to denote that this image is an Example image
disable_sm_tag: False # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
inference:
device_types: ["cpu", "gpu"]
5:
framework: "tensorflow"
version: "2.3.2"
training:
device_types: ["gpu"]
python_versions: ["py37"]
os_version: "ubuntu18.04"
cuda_version: "cu102"
cuda_version: "cu110"
example: False # [Default: False] Set to True to denote that this image is an Example image
disable_sm_tag: True # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
6:
framework: "tensorflow"
version: "2.3.2"
training:
device_types: ["gpu"]
python_versions: ["py37"]
os_version: "ubuntu18.04"
cuda_version: "cu110"
example: True # [Default: False] Set to True to denote that this image is an Example image
disable_sm_tag: False # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
5:
7:
framework: "tensorflow"
version: "1.15.5"
training:
Expand All @@ -88,7 +104,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
6:
8:
framework: "tensorflow"
version: "1.15.5"
training:
Expand All @@ -101,7 +117,7 @@ release_images:
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
7:
9:
framework: "tensorflow"
version: "2.3.0"
inference:
Expand All @@ -113,7 +129,7 @@ release_images:
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
8:
10:
framework: "mxnet"
version: "1.8.0"
training:
Expand All @@ -134,7 +150,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
9:
11:
framework: "mxnet"
version: "1.8.0"
training:
Expand All @@ -145,7 +161,7 @@ release_images:
example: True
disable_sm_tag: False # [Default: False] This option is not used by Example images
force_release: False
10:
12:
framework: "mxnet"
version: "1.7.0"
training:
Expand All @@ -164,7 +180,7 @@ release_images:
example: False
disable_sm_tag: False # [Default: False] This option is not used by Example images
force_release: False
11:
13:
framework: "mxnet"
version: "1.7.0"
training:
Expand All @@ -175,7 +191,7 @@ release_images:
example: True
disable_sm_tag: False # [Default: False] This option is not used by Example images
force_release: False
12:
14:
framework: "tensorflow"
version: "2.4.1"
training:
Expand All @@ -194,7 +210,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
13:
15:
framework: "tensorflow"
version: "2.4.1"
training:
Expand All @@ -205,7 +221,7 @@ release_images:
example: True
disable_sm_tag: False
force_release: False
14:
16:
framework: "pytorch"
version: "1.8.1"
training:
Expand All @@ -228,7 +244,7 @@ release_images:
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
15:
17:
framework: "pytorch"
version: "1.8.1"
training:
Expand All @@ -239,7 +255,7 @@ release_images:
example: True
disable_sm_tag: False # [Default: False] This option is not used by Example images
force_release: False
16:
18:
framework: "pytorch"
version: "1.5.1"
training:
Expand All @@ -258,7 +274,7 @@ release_images:
example: False
disable_sm_tag: False # [Default: False] This option is not used by Example images
force_release: False
17:
19:
framework: "pytorch"
version: "1.5.1"
training:
Expand All @@ -269,32 +285,6 @@ release_images:
example: True
disable_sm_tag: False # [Default: False] This option is not used by Example images
force_release: False
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we would also need example image configuration for GPU CU102

18:
framework: "tensorflow"
version: "2.3.2"
training:
device_types: ["gpu"]
python_versions: ["py37"]
os_version: "ubuntu18.04"
cuda_version: "cu110"
example: False # [Default: False] Set to True to denote that this image is an Example image
disable_sm_tag: True # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
19:
framework: "tensorflow"
version: "2.3.2"
training:
device_types: ["gpu"]
python_versions: ["py37"]
os_version: "ubuntu18.04"
cuda_version: "cu110"
example: True # [Default: False] Set to True to denote that this image is an Example image
disable_sm_tag: False # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
# to images being published.
force_release: False # [Default: False] Set to True to force images to be published even if the same image
# has already been published. Re-released image will have minor version incremented by 1.
20:
framework: "huggingface_tensorflow"
version: "2.4.1"
Expand Down