Skip to content

Commit 970eb4e

Browse files
saimidusaimidu
committed
feature: Add support for new TF versions
1 parent c95c75a commit 970eb4e

File tree

2 files changed

+230
-4
lines changed

2 files changed

+230
-4
lines changed

‎src/sagemaker/fw_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@
5959
"local_gpu",
6060
)
6161
SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {
62-
"tensorflow": ["2.3", "2.3.1", "2.3.2", "2.4", "2.4.1"],
62+
"tensorflow": ["2.3", "2.3.1", "2.3.2", "2.4", "2.4.1", "2.4.3", "2.5", "2.5.0", "2.5.1", "2.6", "2.6.0"],
6363
"pytorch": ["1.6", "1.6.0", "1.7", "1.7.1", "1.8", "1.8.0", "1.8.1", "1.9", "1.9.0"],
6464
}
6565
SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]
@@ -533,7 +533,7 @@ def _validate_smdataparallel_args(
533533
if "py3" not in py_version:
534534
err_msg += (
535535
f"Provided py_version {py_version} is not supported by smdataparallel.\n"
536-
"Please specify py_version=py3"
536+
"Please specify py_version>=py3"
537537
)
538538

539539
if err_msg:

‎src/sagemaker/image_uri_config/tensorflow.json

Lines changed: 228 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,9 @@
277277
"2.1": "2.1.3",
278278
"2.2": "2.2.2",
279279
"2.3": "2.3.2",
280-
"2.4": "2.4.1"
280+
"2.4": "2.4.3",
281+
"2.5": "2.5.1",
282+
"2.6": "2.6.0"
281283
},
282284
"versions": {
283285
"1.10.0": {
@@ -1251,6 +1253,96 @@
12511253
"us-west-2": "763104351884"
12521254
},
12531255
"repository": "tensorflow-inference"
1256+
},
1257+
"2.4.3": {
1258+
"registries": {
1259+
"af-south-1": "626614931356",
1260+
"ap-east-1": "871362719292",
1261+
"ap-northeast-1": "763104351884",
1262+
"ap-northeast-2": "763104351884",
1263+
"ap-northeast-3": "364406365360",
1264+
"ap-south-1": "763104351884",
1265+
"ap-southeast-1": "763104351884",
1266+
"ap-southeast-2": "763104351884",
1267+
"ca-central-1": "763104351884",
1268+
"cn-north-1": "727897471807",
1269+
"cn-northwest-1": "727897471807",
1270+
"eu-central-1": "763104351884",
1271+
"eu-north-1": "763104351884",
1272+
"eu-south-1": "692866216735",
1273+
"eu-west-1": "763104351884",
1274+
"eu-west-2": "763104351884",
1275+
"eu-west-3": "763104351884",
1276+
"me-south-1": "217643126080",
1277+
"sa-east-1": "763104351884",
1278+
"us-east-1": "763104351884",
1279+
"us-east-2": "763104351884",
1280+
"us-gov-west-1": "442386744353",
1281+
"us-iso-east-1": "886529160074",
1282+
"us-west-1": "763104351884",
1283+
"us-west-2": "763104351884"
1284+
},
1285+
"repository": "tensorflow-inference"
1286+
},
1287+
"2.5.1": {
1288+
"registries": {
1289+
"af-south-1": "626614931356",
1290+
"ap-east-1": "871362719292",
1291+
"ap-northeast-1": "763104351884",
1292+
"ap-northeast-2": "763104351884",
1293+
"ap-northeast-3": "364406365360",
1294+
"ap-south-1": "763104351884",
1295+
"ap-southeast-1": "763104351884",
1296+
"ap-southeast-2": "763104351884",
1297+
"ca-central-1": "763104351884",
1298+
"cn-north-1": "727897471807",
1299+
"cn-northwest-1": "727897471807",
1300+
"eu-central-1": "763104351884",
1301+
"eu-north-1": "763104351884",
1302+
"eu-south-1": "692866216735",
1303+
"eu-west-1": "763104351884",
1304+
"eu-west-2": "763104351884",
1305+
"eu-west-3": "763104351884",
1306+
"me-south-1": "217643126080",
1307+
"sa-east-1": "763104351884",
1308+
"us-east-1": "763104351884",
1309+
"us-east-2": "763104351884",
1310+
"us-gov-west-1": "442386744353",
1311+
"us-iso-east-1": "886529160074",
1312+
"us-west-1": "763104351884",
1313+
"us-west-2": "763104351884"
1314+
},
1315+
"repository": "tensorflow-inference"
1316+
},
1317+
"2.6.0": {
1318+
"registries": {
1319+
"af-south-1": "626614931356",
1320+
"ap-east-1": "871362719292",
1321+
"ap-northeast-1": "763104351884",
1322+
"ap-northeast-2": "763104351884",
1323+
"ap-northeast-3": "364406365360",
1324+
"ap-south-1": "763104351884",
1325+
"ap-southeast-1": "763104351884",
1326+
"ap-southeast-2": "763104351884",
1327+
"ca-central-1": "763104351884",
1328+
"cn-north-1": "727897471807",
1329+
"cn-northwest-1": "727897471807",
1330+
"eu-central-1": "763104351884",
1331+
"eu-north-1": "763104351884",
1332+
"eu-south-1": "692866216735",
1333+
"eu-west-1": "763104351884",
1334+
"eu-west-2": "763104351884",
1335+
"eu-west-3": "763104351884",
1336+
"me-south-1": "217643126080",
1337+
"sa-east-1": "763104351884",
1338+
"us-east-1": "763104351884",
1339+
"us-east-2": "763104351884",
1340+
"us-gov-west-1": "442386744353",
1341+
"us-iso-east-1": "886529160074",
1342+
"us-west-1": "763104351884",
1343+
"us-west-2": "763104351884"
1344+
},
1345+
"repository": "tensorflow-inference"
12541346
}
12551347
}
12561348
},
@@ -1276,7 +1368,9 @@
12761368
"2.1": "2.1.3",
12771369
"2.2": "2.2.2",
12781370
"2.3": "2.3.2",
1279-
"2.4": "2.4.1"
1371+
"2.4": "2.4.3",
1372+
"2.5": "2.5.1",
1373+
"2.6": "2.6.0"
12801374
},
12811375
"versions": {
12821376
"1.10.0": {
@@ -2370,6 +2464,138 @@
23702464
"us-west-2": "763104351884"
23712465
},
23722466
"repository": "tensorflow-training"
2467+
},
2468+
"2.4.3": {
2469+
"py_versions": [
2470+
"py37"
2471+
],
2472+
"registries": {
2473+
"af-south-1": "626614931356",
2474+
"ap-east-1": "871362719292",
2475+
"ap-northeast-1": "763104351884",
2476+
"ap-northeast-2": "763104351884",
2477+
"ap-northeast-3": "364406365360",
2478+
"ap-south-1": "763104351884",
2479+
"ap-southeast-1": "763104351884",
2480+
"ap-southeast-2": "763104351884",
2481+
"ca-central-1": "763104351884",
2482+
"cn-north-1": "727897471807",
2483+
"cn-northwest-1": "727897471807",
2484+
"eu-central-1": "763104351884",
2485+
"eu-north-1": "763104351884",
2486+
"eu-south-1": "692866216735",
2487+
"eu-west-1": "763104351884",
2488+
"eu-west-2": "763104351884",
2489+
"eu-west-3": "763104351884",
2490+
"me-south-1": "217643126080",
2491+
"sa-east-1": "763104351884",
2492+
"us-east-1": "763104351884",
2493+
"us-east-2": "763104351884",
2494+
"us-gov-west-1": "442386744353",
2495+
"us-iso-east-1": "886529160074",
2496+
"us-west-1": "763104351884",
2497+
"us-west-2": "763104351884"
2498+
},
2499+
"repository": "tensorflow-training"
2500+
},
2501+
"2.5.0": {
2502+
"py_versions": [
2503+
"py37"
2504+
],
2505+
"registries": {
2506+
"af-south-1": "626614931356",
2507+
"ap-east-1": "871362719292",
2508+
"ap-northeast-1": "763104351884",
2509+
"ap-northeast-2": "763104351884",
2510+
"ap-northeast-3": "364406365360",
2511+
"ap-south-1": "763104351884",
2512+
"ap-southeast-1": "763104351884",
2513+
"ap-southeast-2": "763104351884",
2514+
"ca-central-1": "763104351884",
2515+
"cn-north-1": "727897471807",
2516+
"cn-northwest-1": "727897471807",
2517+
"eu-central-1": "763104351884",
2518+
"eu-north-1": "763104351884",
2519+
"eu-south-1": "692866216735",
2520+
"eu-west-1": "763104351884",
2521+
"eu-west-2": "763104351884",
2522+
"eu-west-3": "763104351884",
2523+
"me-south-1": "217643126080",
2524+
"sa-east-1": "763104351884",
2525+
"us-east-1": "763104351884",
2526+
"us-east-2": "763104351884",
2527+
"us-gov-west-1": "442386744353",
2528+
"us-iso-east-1": "886529160074",
2529+
"us-west-1": "763104351884",
2530+
"us-west-2": "763104351884"
2531+
},
2532+
"repository": "tensorflow-training"
2533+
},
2534+
"2.5.1": {
2535+
"py_versions": [
2536+
"py37"
2537+
],
2538+
"registries": {
2539+
"af-south-1": "626614931356",
2540+
"ap-east-1": "871362719292",
2541+
"ap-northeast-1": "763104351884",
2542+
"ap-northeast-2": "763104351884",
2543+
"ap-northeast-3": "364406365360",
2544+
"ap-south-1": "763104351884",
2545+
"ap-southeast-1": "763104351884",
2546+
"ap-southeast-2": "763104351884",
2547+
"ca-central-1": "763104351884",
2548+
"cn-north-1": "727897471807",
2549+
"cn-northwest-1": "727897471807",
2550+
"eu-central-1": "763104351884",
2551+
"eu-north-1": "763104351884",
2552+
"eu-south-1": "692866216735",
2553+
"eu-west-1": "763104351884",
2554+
"eu-west-2": "763104351884",
2555+
"eu-west-3": "763104351884",
2556+
"me-south-1": "217643126080",
2557+
"sa-east-1": "763104351884",
2558+
"us-east-1": "763104351884",
2559+
"us-east-2": "763104351884",
2560+
"us-gov-west-1": "442386744353",
2561+
"us-iso-east-1": "886529160074",
2562+
"us-west-1": "763104351884",
2563+
"us-west-2": "763104351884"
2564+
},
2565+
"repository": "tensorflow-training"
2566+
},
2567+
"2.6.0": {
2568+
"py_versions": [
2569+
"py38"
2570+
],
2571+
"registries": {
2572+
"af-south-1": "626614931356",
2573+
"ap-east-1": "871362719292",
2574+
"ap-northeast-1": "763104351884",
2575+
"ap-northeast-2": "763104351884",
2576+
"ap-northeast-3": "364406365360",
2577+
"ap-south-1": "763104351884",
2578+
"ap-southeast-1": "763104351884",
2579+
"ap-southeast-2": "763104351884",
2580+
"ca-central-1": "763104351884",
2581+
"cn-north-1": "727897471807",
2582+
"cn-northwest-1": "727897471807",
2583+
"eu-central-1": "763104351884",
2584+
"eu-north-1": "763104351884",
2585+
"eu-south-1": "692866216735",
2586+
"eu-west-1": "763104351884",
2587+
"eu-west-2": "763104351884",
2588+
"eu-west-3": "763104351884",
2589+
"me-south-1": "217643126080",
2590+
"sa-east-1": "763104351884",
2591+
"us-east-1": "763104351884",
2592+
"us-east-2": "763104351884",
2593+
"us-gov-west-1": "442386744353",
2594+
"us-iso-east-1": "886529160074",
2595+
"us-west-1": "763104351884",
2596+
"us-west-2": "763104351884"
2597+
},
2598+
"repository": "tensorflow-training"
23732599
}
23742600
}
23752601
}

0 commit comments

Comments
 (0)