diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py index 0a217388f6..f5547b50bd 100644 --- a/src/sagemaker/fw_utils.py +++ b/src/sagemaker/fw_utils.py @@ -123,6 +123,7 @@ "1.11.0", "1.12", "1.12.0", + "1.12.1", ], } @@ -134,6 +135,7 @@ "1.11.0", "1.12", "1.12.0", + "1.12.1", ] diff --git a/src/sagemaker/image_uri_config/pytorch.json b/src/sagemaker/image_uri_config/pytorch.json index 74127a1fda..3bf8016ba8 100644 --- a/src/sagemaker/image_uri_config/pytorch.json +++ b/src/sagemaker/image_uri_config/pytorch.json @@ -68,7 +68,7 @@ "1.9": "1.9.1", "1.10": "1.10.2", "1.11": "1.11.0", - "1.12": "1.12.0" + "1.12": "1.12.1" }, "versions": { "0.4.0": { @@ -651,6 +651,40 @@ "us-west-2": "763104351884" }, "repository": "pytorch-inference" + }, + "1.12.1": { + "py_versions": [ + "py38" + ], + "registries": { + "af-south-1": "626614931356", + "ap-east-1": "871362719292", + "ap-northeast-1": "763104351884", + "ap-northeast-2": "763104351884", + "ap-northeast-3": "364406365360", + "ap-south-1": "763104351884", + "ap-southeast-1": "763104351884", + "ap-southeast-2": "763104351884", + "ap-southeast-3": "907027046896", + "ca-central-1": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", + "eu-central-1": "763104351884", + "eu-north-1": "763104351884", + "eu-west-1": "763104351884", + "eu-west-2": "763104351884", + "eu-west-3": "763104351884", + "eu-south-1": "692866216735", + "me-south-1": "217643126080", + "sa-east-1": "763104351884", + "us-east-1": "763104351884", + "us-east-2": "763104351884", + "us-gov-west-1": "442386744353", + "us-iso-east-1": "886529160074", + "us-west-1": "763104351884", + "us-west-2": "763104351884" + }, + "repository": "pytorch-inference" } } }, @@ -718,7 +752,7 @@ "1.9": "1.9.1", "1.10": "1.10.2", "1.11": "1.11.0", - "1.12": "1.12.0" + "1.12": "1.12.1" }, "versions": { "0.4.0": { @@ -1302,6 +1336,40 @@ "us-west-2": "763104351884" }, "repository": "pytorch-training" + }, + "1.12.1": { + "py_versions": [ + "py38" + ], + "registries": { + "af-south-1": "626614931356", + "ap-east-1": "871362719292", + "ap-northeast-1": "763104351884", + "ap-northeast-2": "763104351884", + "ap-northeast-3": "364406365360", + "ap-south-1": "763104351884", + "ap-southeast-1": "763104351884", + "ap-southeast-2": "763104351884", + "ap-southeast-3": "907027046896", + "ca-central-1": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", + "eu-central-1": "763104351884", + "eu-north-1": "763104351884", + "eu-west-1": "763104351884", + "eu-west-2": "763104351884", + "eu-west-3": "763104351884", + "eu-south-1": "692866216735", + "me-south-1": "217643126080", + "sa-east-1": "763104351884", + "us-east-1": "763104351884", + "us-east-2": "763104351884", + "us-gov-west-1": "442386744353", + "us-iso-east-1": "886529160074", + "us-west-1": "763104351884", + "us-west-2": "763104351884" + }, + "repository": "pytorch-training" } } } diff --git a/tests/scripts/run-notebook-test.sh b/tests/scripts/run-notebook-test.sh index 528961c338..0407a857f4 100755 --- a/tests/scripts/run-notebook-test.sh +++ b/tests/scripts/run-notebook-test.sh @@ -119,6 +119,7 @@ echo "set SAGEMAKER_ROLE_ARN=$SAGEMAKER_ROLE_ARN" --region us-west-2 \ --lifecycle-config-name $LIFECYCLE_CONFIG_NAME \ --notebook-instance-role-arn $SAGEMAKER_ROLE_ARN \ +--platformIdentifier notebook-al2-v2 \ ./amazon-sagemaker-examples/sagemaker_processing/spark_distributed_data_processing/sagemaker-spark-processing.ipynb \ ./amazon-sagemaker-examples/advanced_functionality/kmeans_bring_your_own_model/kmeans_bring_your_own_model.ipynb \ ./amazon-sagemaker-examples/advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb \ diff --git a/tests/unit/test_fw_utils.py b/tests/unit/test_fw_utils.py index 438c0a6595..204dd3737e 100644 --- a/tests/unit/test_fw_utils.py +++ b/tests/unit/test_fw_utils.py @@ -864,6 +864,7 @@ def test_validate_smdataparallel_args_not_raises(): ("ml.p3.16xlarge", "pytorch", "1.11.0", "py38", smdataparallel_enabled), ("ml.p3.16xlarge", "pytorch", "1.11", "py38", smdataparallel_enabled), ("ml.p3.16xlarge", "pytorch", "1.12.0", "py38", smdataparallel_enabled), + ("ml.p3.16xlarge", "pytorch", "1.12.1", "py38", smdataparallel_enabled), ("ml.p3.16xlarge", "pytorch", "1.12", "py38", smdataparallel_enabled), ("ml.p3.16xlarge", "tensorflow", "2.4.1", "py3", smdataparallel_enabled_custom_mpi), ("ml.p3.16xlarge", "tensorflow", "2.4.1", "py37", smdataparallel_enabled_custom_mpi), @@ -882,6 +883,7 @@ def test_validate_smdataparallel_args_not_raises(): ("ml.p3.16xlarge", "pytorch", "1.10.2", "py38", smdataparallel_enabled_custom_mpi), ("ml.p3.16xlarge", "pytorch", "1.11.0", "py38", smdataparallel_enabled_custom_mpi), ("ml.p3.16xlarge", "pytorch", "1.12.0", "py38", smdataparallel_enabled_custom_mpi), + ("ml.p3.16xlarge", "pytorch", "1.12.1", "py38", smdataparallel_enabled_custom_mpi), ] for instance_type, framework_name, framework_version, py_version, distribution in good_args: fw_utils._validate_smdataparallel_args( @@ -917,6 +919,7 @@ def test_validate_pytorchddp_not_raises(): "1.11.0", "1.12", "1.12.0", + "1.12.1", ] for framework_version in pytorchddp_supported_fw_versions: fw_utils.validate_pytorch_distribution(