Skip to content

Commit 68eacf2

Browse files
author
Rohit Kumar Srivastava
committed
removing gov cloud regions and making pytorch container cuda-11.3
1 parent f2e961f commit 68eacf2

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

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

+3-7
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
"pytorch1.10.2": {
4343
"py_versions": ["py38"],
4444
"registries": {
45-
"af-south-1": "626614931356",
45+
"af-south-1": "626614931356",
4646
"ap-east-1": "871362719292",
4747
"ap-northeast-1": "763104351884",
4848
"ap-northeast-2": "763104351884",
@@ -61,18 +61,16 @@
6161
"sa-east-1": "763104351884",
6262
"us-east-1": "763104351884",
6363
"us-east-2": "763104351884",
64-
"us-gov-west-1": "442386744353",
65-
"us-iso-east-1": "886529160074",
6664
"us-west-1": "763104351884",
6765
"us-west-2": "763104351884"
6866
},
6967
"repository": "huggingface-pytorch-trcomp-training",
70-
"container_version": {"gpu":"cu112-ubuntu20.04"}
68+
"container_version": {"gpu":"cu113-ubuntu20.04"}
7169
},
7270
"tensorflow2.6.3": {
7371
"py_versions": ["py38"],
7472
"registries": {
75-
"af-south-1": "626614931356",
73+
"af-south-1": "626614931356",
7674
"ap-east-1": "871362719292",
7775
"ap-northeast-1": "763104351884",
7876
"ap-northeast-2": "763104351884",
@@ -91,8 +89,6 @@
9189
"sa-east-1": "763104351884",
9290
"us-east-1": "763104351884",
9391
"us-east-2": "763104351884",
94-
"us-gov-west-1": "442386744353",
95-
"us-iso-east-1": "886529160074",
9692
"us-west-1": "763104351884",
9793
"us-west-2": "763104351884"
9894
},

0 commit comments

Comments
 (0)