Instalar el SDK de Vertex AI para Python

Usa el SDK de Vertex AI para Python para automatizar tus flujos de trabajo de aprendizaje automático (ML). En este tema se explica cómo instalar el SDK de Vertex AI para Python. Para obtener más información sobre el SDK de Vertex AI, consulta los siguientes recursos:

La instalación del SDK de Vertex AI para Python incluye los siguientes pasos:

  1. Crear un entorno de Python aislado
  2. Instalar el paquete del SDK de Vertex AI
  3. Inicializar el SDK de Vertex AI

Crear un entorno de Python aislado

Una práctica recomendada de Python es instalar el SDK de Vertex AI en un entorno de Python aislado para cada proyecto. De esta forma, se evitan conflictos de dependencias, versiones y permisos. Puedes crear un entorno aislado para usar la línea de comandos en un shell o para usar un cuaderno.

Para crear un entorno aislado cuando uses la línea de comandos, activa un entorno venv. Una vez que se haya activado el entorno venv, podrá instalar el SDK de Vertex AI y ejecutar sus secuencias de comandos de Python. Para obtener más información, consulta Usar venv para aislar dependencias y Configurar un entorno de desarrollo de Python.

Para usar un cuaderno en un entorno aislado, puedes crear una instancia de Vertex AI Workbench. A continuación, instala el SDK de Vertex AI y ejecuta tus secuencias de comandos de Python desde un cuaderno de tu instancia de Vertex AI Workbench. Para obtener más información, consulta Crear una instancia de Vertex AI Workbench.

Instalar o actualizar el paquete del SDK de Vertex AI

Para instalar o actualizar el SDK de Vertex AI, ejecuta el siguiente comando en tu entorno virtual:

pip install --upgrade google-cloud-aiplatform

Inicializar el SDK de Vertex AI

Después de instalar el SDK de Vertex AI para Python, debes inicializarlo con los detalles de tu cuenta de Vertex AI y Google Cloud . Por ejemplo, cuando inicializas el SDK, especificas información como el nombre del proyecto, la región y el segmento de Cloud Storage de staging. El siguiente método es un ejemplo de un método que inicializa el SDK de 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,
    )

Siguientes pasos