|
18 | 18 | from e2e.bootstrap_resources import get_bootstrap_resources
|
19 | 19 |
|
20 | 20 | # Taken from the SageMaker Python SDK
|
| 21 | +# https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/image_uri_config/xgboost.json |
21 | 22 | # Rather than including the entire SDK
|
22 | 23 | XGBOOST_IMAGE_URIS = {
|
23 | 24 | "us-west-1": "746614075791.dkr.ecr.us-west-1.amazonaws.com",
|
|
40 | 41 | "eu-west-3": "659782779980.dkr.ecr.eu-west-3.amazonaws.com",
|
41 | 42 | "me-south-1": "801668240914.dkr.ecr.me-south-1.amazonaws.com",
|
42 | 43 | "sa-east-1": "737474898029.dkr.ecr.sa-east-1.amazonaws.com",
|
| 44 | + "eu-south-1": "978288397137.dkr.ecr.eu-south-1.amazonaws.com", |
| 45 | + "ap-northeast-3": "867004704886.dkr.ecr.ap-northeast-3.amazonaws.com", |
| 46 | + "eu-south-2": "104374241257.dkr.ecr.eu-south-2.amazonaws.com", |
| 47 | + "ap-southeast-3": "951798379941.dkr.ecr.ap-southeast-3.amazonaws.com", |
| 48 | + "me-central-1": "272398656194.dkr.ecr.me-central-1.amazonaws.com", |
| 49 | + "eu-central-2": "680994064768.dkr.ecr.eu-central-2.amazonaws.com", |
43 | 50 | }
|
44 | 51 |
|
| 52 | +# https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/image_uri_config/debugger.json |
45 | 53 | DEBUGGER_IMAGE_URIS = {
|
46 | 54 | "us-west-1": "685455198987.dkr.ecr.us-west-1.amazonaws.com",
|
47 | 55 | "us-west-2": "895741380848.dkr.ecr.us-west-2.amazonaws.com",
|
|
63 | 71 | "eu-west-3": "447278800020.dkr.ecr.eu-west-3.amazonaws.com",
|
64 | 72 | "me-south-1": "986000313247.dkr.ecr.me-south-1.amazonaws.com",
|
65 | 73 | "sa-east-1": "818342061345.dkr.ecr.sa-east-1.amazonaws.com",
|
| 74 | + "eu-south-1": "563282790590.dkr.ecr.eu-south-1.amazonaws.com", |
| 75 | + "ap-northeast-3": "479947661362.dkr.ecr.ap-northeast-3.amazonaws.com", |
66 | 76 | }
|
67 | 77 |
|
68 | 78 | # https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
|
|
87 | 97 | "eu-west-3": "749696950732.dkr.ecr.eu-west-3.amazonaws.com",
|
88 | 98 | "me-south-1": "249704162688.dkr.ecr.me-south-1.amazonaws.com",
|
89 | 99 | "sa-east-1": "855470959533.dkr.ecr.sa-east-1.amazonaws.com",
|
| 100 | + "eu-south-1": "257386234256.dkr.ecr.eu-south-1.amazonaws.com", |
| 101 | + "ap-northeast-3": "867004704886.dkr.ecr.ap-northeast-3.amazonaws.com", |
| 102 | + "eu-south-2": "104374241257.dkr.ecr.eu-south-2.amazonaws.com", |
| 103 | + "ap-southeast-3": "951798379941.dkr.ecr.ap-southeast-3.amazonaws.com", |
| 104 | + "me-central-1": "272398656194.dkr.ecr.me-central-1.amazonaws.com", |
| 105 | + "eu-central-2": "680994064768.dkr.ecr.eu-central-2.amazonaws.com", |
90 | 106 | }
|
91 | 107 |
|
92 | 108 |
|
|
113 | 129 | "sa-east-1": "763104351884.dkr.ecr.sa-east-1.amazonaws.com",
|
114 | 130 | "cn-north-1": "727897471807.dkr.ecr.cn-north-1.amazonaws.com.cn",
|
115 | 131 | "cn-northwest-1": "727897471807.dkr.ecr.cn-northwest-1.amazonaws.com.cn",
|
| 132 | + "ap-northeast-3": "364406365360.dkr.ecr.ap-northeast-3.amazonaws.com", |
| 133 | + "eu-south-2": "503227376785.dkr.ecr.eu-south-2.amazonaws.com", |
| 134 | + "ap-southeast-3": "907027046896.dkr.ecr.ap-southeast-3.amazonaws.com", |
| 135 | + "me-central-1": "914824155844.dkr.ecr.me-central-1.amazonaws.com", |
| 136 | + "eu-central-2": "380420809688.dkr.ecr.eu-central-2.amazonaws.com", |
116 | 137 | }
|
117 | 138 |
|
118 | 139 | # https://docs.aws.amazon.com/sagemaker/latest/dg/model-monitor-pre-built-container.html
|
|
140 | 161 | "me-south-1": "607024016150.dkr.ecr.me-south-1.amazonaws.com",
|
141 | 162 | "sa-east-1": "539772159869.dkr.ecr.sa-east-1.amazonaws.com",
|
142 | 163 | "us-gov-west-1": "362178532790.dkr.ecr.us-gov-west-1.amazonaws.com",
|
| 164 | + "ap-northeast-3": "990339680094.dkr.ecr.ap-northeast-3.amazonaws.com", |
| 165 | + "ap-southeast-3": "669540362728.dkr.ecr.ap-southeast-3.amazonaws.com", |
143 | 166 | }
|
144 | 167 |
|
145 | 168 | # https://docs.aws.amazon.com/sagemaker/latest/dg/clarify-configure-processing-jobs.html#clarify-processing-job-configure-container
|
|
164 | 187 | "sa-east-1": "520018980103.dkr.ecr.sa-east-1.amazonaws.com",
|
165 | 188 | "af-south-1": "811711786498.dkr.ecr.af-south-1.amazonaws.com",
|
166 | 189 | "eu-south-1": "638885417683.dkr.ecr.eu-south-1.amazonaws.com",
|
| 190 | + "ap-northeast-3": "912233562940.dkr.ecr.ap-northeast-3.amazonaws.com", |
| 191 | + "ap-southeast-3": "705930551576.dkr.ecr.ap-southeast-3.amazonaws.com", |
167 | 192 | }
|
168 | 193 |
|
169 | 194 | ENDPOINT_INSTANCE_TYPES = {
|
|
174 | 199 | TRAINING_JOB_INSTANCE_TYPES = {
|
175 | 200 | "eu-west-3": "ml.m5.xlarge",
|
176 | 201 | "eu-north-1": "ml.m5.xlarge",
|
| 202 | + "me-south-1": "ml.m5.xlarge", |
177 | 203 | }
|
178 | 204 |
|
179 | 205 | NOTEBOOK_INSTANCE_INSTANCE_TYPES = {
|
|
0 commit comments