Skip to content

Commit

Permalink
[PyTorch][Inference][EC2] Upgrade torchserve to 0.12.0 in PT 2.4 (#4401)
Browse files Browse the repository at this point in the history
* Upgrade TS to 0.12.0 and disable autopatch

* Update ['dlc_developer_config.toml']

dlc_developer_config.toml:
{   'build': {   'build_frameworks': ['pytorch'],
                 'build_inference': True,
                 'build_training': False},
    'buildspec_override': {   'dlc-pr-pytorch-inference': 'pytorch/inference/buildspec-2-4-ec2.yml'},
    'dev': {   'deep_canary_mode': False,
               'graviton_mode': False,
               'neuronx_mode': False},
    'test': {   'ec2_tests': True,
                'ecs_tests': True,
                'eks_tests': True,
                'sagemaker_local_tests': True,
                'sagemaker_remote_tests': True,
                'sanity_tests': True}}

* update core_packages for gpu images

* Update ['dlc_developer_config.toml']

dlc_developer_config.toml:
{   'build': {   'build_frameworks': ['pytorch'],
                 'build_inference': True,
                 'build_training': False},
    'buildspec_override': {   'dlc-pr-pytorch-graviton-inference': 'pytorch/inference/buildspec-graviton-2-4-ec2.yml'},
    'dev': {   'deep_canary_mode': False,
               'graviton_mode': True,
               'neuronx_mode': False},
    'test': {   'ec2_tests': True,
                'ecs_tests': True,
                'eks_tests': True,
                'sagemaker_local_tests': True,
                'sagemaker_remote_tests': True,
                'sanity_tests': True}}

* Restore ['dlc_developer_config.toml']

dlc_developer_config.toml:
('Restore to '
 'https://raw.githubusercontent.com/aws/deep-learning-containers/master/dlc_developer_config.toml')
  • Loading branch information
sallyseok authored Nov 1, 2024
1 parent ce72b13 commit f97413f
Show file tree
Hide file tree
Showing 6 changed files with 14 additions and 14 deletions.
6 changes: 3 additions & 3 deletions pytorch/inference/buildspec-2-4-ec2.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
version: &VERSION 2.4.0
short_version: &SHORT_VERSION "2.4"
arch_type: x86
autopatch_build: "True"
# autopatch_build: "True"

repository_info:
inference_repository: &INFERENCE_REPOSITORY
Expand Down Expand Up @@ -40,7 +40,7 @@ images:
python_version: &DOCKER_PYTHON_VERSION py3
tag_python_version: &TAG_PYTHON_VERSION py311
os_version: &OS_VERSION ubuntu22.04
torch_serve_version: &TORCHSERVE_VERSION 0.11.1
torch_serve_version: &TORCHSERVE_VERSION 0.12.0
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-ec2" ]
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-ec2" ]
# build_tag_override: "beta:2.4.0-cpu-py311-ubuntu22.04-ec2"
Expand All @@ -57,7 +57,7 @@ images:
tag_python_version: &TAG_PYTHON_VERSION py311
cuda_version: &CUDA_VERSION cu124
os_version: &OS_VERSION ubuntu22.04
torch_serve_version: &TORCHSERVE_VERSION 0.11.1
torch_serve_version: &TORCHSERVE_VERSION 0.12.0
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-ec2" ]
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-ec2" ]
# build_tag_override: "beta:2.4.0-gpu-py311-cu124-ubuntu22.04-ec2"
Expand Down
6 changes: 3 additions & 3 deletions pytorch/inference/buildspec-graviton-2-4-ec2.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
version: &VERSION 2.4.0
short_version: &SHORT_VERSION "2.4"
arch_type: graviton
autopatch_build: "True"
# autopatch_build: "True"

repository_info:
inference_repository: &INFERENCE_REPOSITORY
Expand Down Expand Up @@ -40,7 +40,7 @@ images:
python_version: &DOCKER_PYTHON_VERSION py3
tag_python_version: &TAG_PYTHON_VERSION py311
os_version: &OS_VERSION ubuntu22.04
torch_serve_version: &TORCHSERVE_VERSION 0.11.1
torch_serve_version: &TORCHSERVE_VERSION 0.12.0
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION , "-ec2"]
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-ec2" ]
# build_tag_override: "beta:2.4.0-cpu-py311-ubuntu22.04-ec2"
Expand All @@ -57,7 +57,7 @@ images:
tag_python_version: &TAG_PYTHON_VERSION py311
cuda_version: &CUDA_VERSION cu124
os_version: &OS_VERSION ubuntu22.04
torch_serve_version: &TORCHSERVE_VERSION 0.11.1
torch_serve_version: &TORCHSERVE_VERSION 0.12.0
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION , "-ec2"]
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-ec2" ]
# build_tag_override: "beta:2.4.0-gpu-py311-cu124-ubuntu22.04-ec2"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,9 @@
"version_specifier": "<2"
},
"torchserve": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
},
"torch-model-archiver": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,9 @@
"version_specifier": ">=2.32.3"
},
"torchserve": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
},
"torch-model-archiver": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,9 @@
"version_specifier": "<2"
},
"torchserve": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
},
"torch-model-archiver": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,9 @@
"version_specifier": ">=2.32.3"
},
"torchserve": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
},
"torch-model-archiver": {
"version_specifier": "==0.11.1"
"version_specifier": "==0.12.0"
}
}

0 comments on commit f97413f

Please sign in to comment.