Instalar o SDK da Vertex AI para Python

Use o SDK da Vertex AI para Python para automatizar seus fluxos de trabalho de machine learning (ML). Neste tópico, mostramos como instalar o SDK da Vertex AI para Python. Para mais informações sobre o SDK da Vertex AI, consulte os seguintes recursos:

A instalação do SDK Vertex AI para Python inclui as seguintes etapas:

  1. Crie um ambiente Python isolado:
  2. Instalar o pacote do SDK da Vertex AI
  3. Inicialize o SDK da Vertex AI

Crie um ambiente Python isolado:

Uma prática recomendada do Python é instalar o SDK Vertex AI em um ambiente Python isolado para cada projeto. Isso ajuda a evitar conflitos de dependência, versão e permissões. É possível criar um ambiente isolado para usar a linha de comando em um shell ou para usar um notebook.

Para criar um ambiente isolado ao usar a linha de comando, ative um ambiente venv . Depois que o ambiente venv for ativado, você poderá instalar o SDK da Vertex AI e executar os scripts do Python. Para ver mais informações, consulte Usar venv para isolar dependências e Configurar um ambiente de desenvolvimento em Python.

Para usar um notebook no ambiente isolado, crie uma instância de notebook. Depois de criar a instância de notebook, use-a para instalar o SDK Vertex AI e executar os scripts Python. Para mais informações, consulte Criar uma instância de notebooks gerenciados pelo usuário.

Instalar ou atualizar o pacote do SDK da Vertex AI

Para instalar ou atualizar o SDK da Vertex AI, execute o seguinte comando no seu ambiente virtual:

pip install --upgrade google-cloud-aiplatform

Inicializar o SDK da Vertex AI

Depois de instalar o SDK da Vertex AI para Python, você precisa inicializar o SDK com os detalhes da Vertex AI e do Google Cloud. Por exemplo, ao inicializar o SDK, você especifica informações como nome, região e bucket de preparo do Cloud Storage para seu projeto. O método a seguir é um exemplo de método que inicializa o SDK da Vertex AI.

def init_sample(
    project: Optional[str] = None,
    location: Optional[str] = None,
    experiment: Optional[str] = None,
    staging_bucket: Optional[str] = None,
    credentials: Optional[google.auth.credentials.Credentials] = None,
    encryption_spec_key_name: Optional[str] = None,
    service_account: Optional[str] = None,
):

    import vertexai

    vertexai.init(
        project=project,
        location=location,
        experiment=experiment,
        staging_bucket=staging_bucket,
        credentials=credentials,
        encryption_spec_key_name=encryption_spec_key_name,
        service_account=service_account,
    )

A seguir