diff --git a/src/sagemaker/image_uri_config/huggingface.json b/src/sagemaker/image_uri_config/huggingface.json index 4907299cb8..48409540c6 100644 --- a/src/sagemaker/image_uri_config/huggingface.json +++ b/src/sagemaker/image_uri_config/huggingface.json @@ -561,7 +561,8 @@ }, "4.17.0": { "version_aliases": { - "pytorch1.10": "pytorch1.10.2" + "pytorch1.10": "pytorch1.10.2", + "tensorflow2.6": "tensorflow2.6.3" }, "pytorch1.10.2": { "py_versions": ["py38"], @@ -594,6 +595,38 @@ }, "repository": "huggingface-pytorch-training", "container_version": {"gpu": "cu113-ubuntu20.04"} + }, + "tensorflow2.6.3": { + "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", + "ca-central-1": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", + "eu-central-1": "763104351884", + "eu-north-1": "763104351884", + "eu-south-1": "692866216735", + "eu-west-1": "763104351884", + "eu-west-2": "763104351884", + "eu-west-3": "763104351884", + "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": "huggingface-tensorflow-training", + "container_version": {"gpu": "cu112-ubuntu20.04"} } } } @@ -989,7 +1022,8 @@ }, "4.17.0": { "version_aliases": { - "pytorch1.10": "pytorch1.10.2" + "pytorch1.10": "pytorch1.10.2", + "tensorflow2.6": "tensorflow2.6.3" }, "pytorch1.10.2": { "py_versions": ["py38"], @@ -1022,6 +1056,38 @@ }, "repository": "huggingface-pytorch-inference", "container_version": {"gpu": "cu113-ubuntu20.04", "cpu": "ubuntu20.04" } + }, + "tensorflow2.6.3": { + "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", + "ca-central-1": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", + "eu-central-1": "763104351884", + "eu-north-1": "763104351884", + "eu-south-1": "692866216735", + "eu-west-1": "763104351884", + "eu-west-2": "763104351884", + "eu-west-3": "763104351884", + "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": "huggingface-tensorflow-inference", + "container_version": {"gpu": "cu112-ubuntu20.04", "cpu": "ubuntu20.04" } } } } diff --git a/tests/conftest.py b/tests/conftest.py index aaf646cf9d..0e818d0106 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -266,7 +266,7 @@ def huggingface_pytorch_latest_inference_py_version(huggingface_inference_pytorc @pytest.fixture(scope="module") def huggingface_tensorflow_latest_training_py_version(): - return "py37" + return "py38" @pytest.fixture(scope="module")