Skip to content

Commit f2e961f

Browse files
author
Rohit Kumar Srivastava
committed
update tf version for training compiler and expand to regular regions
1 parent 30b4ce2 commit f2e961f

File tree

1 file changed

+68
-1
lines changed

1 file changed

+68
-1
lines changed

src/sagemaker/image_uri_config/huggingface-training-compiler.json

+68-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@
22
"training": {
33
"processors": ["gpu"],
44
"version_aliases": {
5-
"4.11": "4.11.0"
5+
"4.11": "4.11.0",
6+
"4.17": "4.17.0"
67
},
78
"versions": {
89
"4.11.0": {
@@ -32,6 +33,72 @@
3233
"repository": "huggingface-tensorflow-trcomp-training",
3334
"container_version": {"gpu":"cu112-ubuntu18.04"}
3435
}
36+
},
37+
"4.17.0": {
38+
"version_aliases": {
39+
"pytorch1.10.2": "pytorch1.10.2",
40+
"tensorflow2.6": "tensorflow2.6.3"
41+
},
42+
"pytorch1.10.2": {
43+
"py_versions": ["py38"],
44+
"registries": {
45+
"af-south-1": "626614931356",
46+
"ap-east-1": "871362719292",
47+
"ap-northeast-1": "763104351884",
48+
"ap-northeast-2": "763104351884",
49+
"ap-northeast-3": "364406365360",
50+
"ap-south-1": "763104351884",
51+
"ap-southeast-1": "763104351884",
52+
"ap-southeast-2": "763104351884",
53+
"ca-central-1": "763104351884",
54+
"eu-central-1": "763104351884",
55+
"eu-north-1": "763104351884",
56+
"eu-south-1": "692866216735",
57+
"eu-west-1": "763104351884",
58+
"eu-west-2": "763104351884",
59+
"eu-west-3": "763104351884",
60+
"me-south-1": "217643126080",
61+
"sa-east-1": "763104351884",
62+
"us-east-1": "763104351884",
63+
"us-east-2": "763104351884",
64+
"us-gov-west-1": "442386744353",
65+
"us-iso-east-1": "886529160074",
66+
"us-west-1": "763104351884",
67+
"us-west-2": "763104351884"
68+
},
69+
"repository": "huggingface-pytorch-trcomp-training",
70+
"container_version": {"gpu":"cu112-ubuntu20.04"}
71+
},
72+
"tensorflow2.6.3": {
73+
"py_versions": ["py38"],
74+
"registries": {
75+
"af-south-1": "626614931356",
76+
"ap-east-1": "871362719292",
77+
"ap-northeast-1": "763104351884",
78+
"ap-northeast-2": "763104351884",
79+
"ap-northeast-3": "364406365360",
80+
"ap-south-1": "763104351884",
81+
"ap-southeast-1": "763104351884",
82+
"ap-southeast-2": "763104351884",
83+
"ca-central-1": "763104351884",
84+
"eu-central-1": "763104351884",
85+
"eu-north-1": "763104351884",
86+
"eu-south-1": "692866216735",
87+
"eu-west-1": "763104351884",
88+
"eu-west-2": "763104351884",
89+
"eu-west-3": "763104351884",
90+
"me-south-1": "217643126080",
91+
"sa-east-1": "763104351884",
92+
"us-east-1": "763104351884",
93+
"us-east-2": "763104351884",
94+
"us-gov-west-1": "442386744353",
95+
"us-iso-east-1": "886529160074",
96+
"us-west-1": "763104351884",
97+
"us-west-2": "763104351884"
98+
},
99+
"repository": "huggingface-tensorflow-trcomp-training",
100+
"container_version": {"gpu":"cu112-ubuntu20.04"}
101+
}
35102
}
36103
}
37104
}

0 commit comments

Comments
 (0)