Instalar el SDK de Vertex AI para Python

Usa el SDK de Vertex AI para Python para automatizar los flujos de trabajo de aprendizaje automático (AA). En este tema, se muestra 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. Crea un entorno aislado de Python
  2. Instala el paquete del SDK de Vertex AI
  3. Inicializa el SDK de Vertex AI

Crea un entorno aislado de Python

Una práctica recomendada de Python es instalar el SDK de Vertex AI en un entorno aislado de Python para cada proyecto. Esto ayuda a evitar conflictos de dependencia, versión y permisos. Puedes crear un entorno aislado para usar la línea de comandos en una shell o para usar un notebook.

Para crear un entorno aislado cuando usas la línea de comandos, activa un entorno venv. Después de activar el entorno venv, estás listo para instalar el SDK de Vertex AI y ejecutar tus secuencias de comandos de Python. Si deseas obtener más información, consulta Usa venv para aislar dependencias y Configura un entorno de desarrollo de Python.

Para usar un notebook para tu entorno aislado, crea una instancia de notebook. Después de crear la instancia de notebook, úsala para instalar el SDK de Vertex AI y ejecutar tus secuencias de comandos de Python. Para obtener más información, consulta Crea una instancia de notebooks administrada por el usuario.

Instala o actualiza 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

Inicializa el SDK de Vertex AI

Después de instalar el SDK de Vertex AI para Python, debes inicializar el SDK con tus detalles de Vertex AI y Google Cloud. Por ejemplo, cuando inicializas el SDK, especificas información como el nombre de tu proyecto, la región y el bucket de etapa de pruebas de Cloud Storage. 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,
    )

¿Qué sigue?