Skip to content

Commit 7c9758f

Browse files
authored
Patch CUDA Compat mounting for TF inference (#4567)
* Patch CUDA Compat mounting for TF inference * build test tf 2.18 * formatting * update pip versino pin * fix pip * test 216 sm * remove dlc.py * add allowlist * fix allowlist * add back deep_learning_containers.py * test no entrypoint * fix allowlist 216 * build test tf 2.18 * fix allowlist 218 * add entrypoint * add entrypoint * test without deep_learning_conatainers.py * use dockerd_entrypoint.sh" * correct chmod * print telemetry logs * run logger * add more logging * use bin env * preserve quoting * test 218 * build test 216 * revert toml
1 parent 1dad848 commit 7c9758f

File tree

9 files changed

+1989
-167
lines changed

9 files changed

+1989
-167
lines changed

tensorflow/inference/buildspec-2-16-sm.yml

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK tensorflow
55
version: &VERSION 2.16.1
66
short_version: &SHORT_VERSION 2.16
77
arch_type: x86
8-
autopatch_build: "True"
8+
# autopatch_build: "True"
99

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY
@@ -19,6 +19,12 @@ repository_info:
1919

2020
context:
2121
inference_context: &INFERENCE_CONTEXT
22+
start_cuda_compat:
23+
source: docker/build_artifacts/start_cuda_compat.sh
24+
target: start_cuda_compat.sh
25+
dockerd_entrypoint:
26+
source: docker/build_artifacts/dockerd_entrypoint.sh
27+
target: dockerd_entrypoint.sh
2228
sagemaker_package_name:
2329
source: docker/build_artifacts/sagemaker
2430
target: sagemaker
@@ -43,8 +49,7 @@ images:
4349
tag_python_version: &TAG_PYTHON_VERSION py310
4450
os_version: &OS_VERSION ubuntu20.04
4551
tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
46-
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION,
47-
"-sagemaker" ]
52+
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
4853
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /Dockerfile., *DEVICE_TYPE ]
4954
target: sagemaker
5055
enable_test_promotion: true
@@ -60,12 +65,9 @@ images:
6065
tag_python_version: &TAG_PYTHON_VERSION py310
6166
cuda_version: &CUDA_VERSION cu122
6267
os_version: &OS_VERSION ubuntu20.04
63-
tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION,
64-
"-sagemaker" ]
65-
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION,
66-
"-", *OS_VERSION, "-sagemaker" ]
67-
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /, *CUDA_VERSION, /Dockerfile.,
68-
*DEVICE_TYPE ]
68+
tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
69+
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
70+
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /, *CUDA_VERSION, /Dockerfile., *DEVICE_TYPE ]
6971
target: sagemaker
7072
enable_test_promotion: true
7173
context:

tensorflow/inference/buildspec-2-18-sm.yml

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK tensorflow
55
version: &VERSION 2.18.0
66
short_version: &SHORT_VERSION 2.18
77
arch_type: x86
8-
autopatch_build: "True"
8+
# autopatch_build: "True"
99

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY
@@ -19,6 +19,12 @@ repository_info:
1919

2020
context:
2121
inference_context: &INFERENCE_CONTEXT
22+
start_cuda_compat:
23+
source: docker/build_artifacts/start_cuda_compat.sh
24+
target: start_cuda_compat.sh
25+
dockerd_entrypoint:
26+
source: docker/build_artifacts/dockerd_entrypoint.sh
27+
target: dockerd_entrypoint.sh
2228
sagemaker_package_name:
2329
source: docker/build_artifacts/sagemaker
2430
target: sagemaker
@@ -43,8 +49,7 @@ images:
4349
tag_python_version: &TAG_PYTHON_VERSION py310
4450
os_version: &OS_VERSION ubuntu20.04
4551
tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
46-
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION,
47-
"-sagemaker" ]
52+
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
4853
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /Dockerfile., *DEVICE_TYPE ]
4954
target: sagemaker
5055
enable_test_promotion: true
@@ -60,12 +65,9 @@ images:
6065
tag_python_version: &TAG_PYTHON_VERSION py310
6166
cuda_version: &CUDA_VERSION cu122
6267
os_version: &OS_VERSION ubuntu20.04
63-
tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION,
64-
"-sagemaker" ]
65-
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION,
66-
"-", *OS_VERSION, "-sagemaker" ]
67-
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /, *CUDA_VERSION, /Dockerfile.,
68-
*DEVICE_TYPE ]
68+
tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
69+
latest_release_tag: !join [ *FRAMEWORK_VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
70+
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /, *CUDA_VERSION, /Dockerfile., *DEVICE_TYPE ]
6971
target: sagemaker
7072
enable_test_promotion: true
7173
context:

0 commit comments

Comments
 (0)