gpt4 book ai didi

python - 如何使用 Python 删除部署到 Unified AI Platform 端点的模型?

转载 作者:行者123 更新时间:2023-12-01 23:25:10 25 4
gpt4 key购买 nike

我已经在 Unified Cloud AI Platform 上成功创建了一个端点,并向其部署了两个 Model - Model BModel C分别占流量的 20% 和 80%。我有第三个 Model - Model A 我之前部署过,现在流量为 0%。

现在,在 Cloud Console(用户界面)上,我可以选择Undeploy 这个 Model A,我可以成功地这样做。但是,我无法弄清楚如何使用 Python 客户端 API 执行此操作。

here 上提供的文档不足以理解我们如何做到这一点。任何帮助将不胜感激。

最佳答案

AI Platform Unified 的文档还没有关于如何在端点中取消部署模型的示例。可以引用AI platform unified rpc reference关于目前可用的服务。这是相关代码:

注意:不要忘记在运行代码之前更新 end_point(端点 ID)、project(项目 ID)和 model_id 的值。

from google.cloud import aiplatform
from google.cloud import aiplatform_v1


def undeploy_model_in_endpoint(
end_point: str,
project: str,
model_id: str,
location: str = "us-central1",
api_endpoint: str = "us-central1-aiplatform.googleapis.com",
timeout: int = 7200,
):
# The AI Platform services require regional API endpoints.
client_options = {"api_endpoint": api_endpoint}
# Initialize client that will be used to create and send requests.
# This client only needs to be created once, and can be reused for multiple requests.
client = aiplatform.gapic.EndpointServiceClient(client_options=client_options)
client_model = aiplatform_v1.services.model_service.ModelServiceClient(client_options=client_options)

# Get deployed_model_id
model_name = f'projects/{project}/locations/{location}/models/{model_id}'
model_request = aiplatform_v1.types.GetModelRequest(name=model_name)
model_info = client_model.get_model(request=model_request)
deployed_models_info = model_info.deployed_models
deployed_model_id=model_info.deployed_models[0].deployed_model_id

name=f'projects/{project}/locations/{location}/endpoints/{end_point}'

undeploy_request = aiplatform_v1.types.UndeployModelRequest(endpoint=name,deployed_model_id=deployed_model_id)
client.undeploy_model(request=undeploy_request)

undeploy_model_in_endpoint(end_point='your-endpoint-id',project='your-project-id',model_id='your-model-id')

关于python - 如何使用 Python 删除部署到 Unified AI Platform 端点的模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67397169/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com