Instale o SDK Vertex AI para Python

Use o SDK Vertex AI para Python para automatizar os seus fluxos de trabalho de aprendizagem automática (ML). Este tópico mostra como instalar o SDK Vertex AI para Python. Para mais informações sobre o SDK Vertex AI, consulte os seguintes recursos:

A instalação do SDK do Vertex AI para Python inclui os seguintes passos:

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

Crie um ambiente Python isolado

Uma prática recomendada do Python é instalar o SDK do Vertex AI num ambiente Python isolado para cada projeto. Isto ajuda a evitar conflitos de dependências, versões e autorizações. Pode criar um ambiente isolado para usar a linha de comandos numa shell ou para usar um bloco de notas.

Para criar um ambiente isolado quando usa a linha de comandos, ative um ambiente venv. Depois de ativar o ambiente venv, pode instalar o SDK do Vertex AI e executar os seus scripts Python. Para mais informações, consulte os artigos Use venv to isolate dependencies e Configure um ambiente de desenvolvimento Python.

Para usar um notebook num ambiente isolado, pode criar uma instância do Vertex AI Workbench. Em seguida, instale o SDK do Vertex AI e execute os seus scripts Python a partir de um notebook na instância do Vertex AI Workbench. Para mais informações, consulte o artigo Crie uma instância do Vertex AI Workbench.

Instale ou atualize o pacote do SDK do Vertex AI

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

pip install --upgrade google-cloud-aiplatform

Inicialize o SDK Vertex AI

Depois de instalar o SDK Vertex AI para Python, tem de inicializar o SDK com os detalhes do Vertex AI e do Google Cloud . Por exemplo, quando inicializa o SDK, especifica informações como o nome do projeto, a região e o contentor do Cloud Storage de preparação. O método seguinte é um exemplo de um método que inicializa o SDK do 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,
    )

O que se segue?