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:

• Si deseas obtener información sobre el SDK de Vertex AI para Python, consulta Introducción al SDK de Vertex AI para Python.
• Si deseas obtener información sobre cómo entrenar un modelo con el SDK de Vertex AI para Python, consulta Entrena un modelo con Vertex AI y el SDK de Python.
• Si deseas obtener información sobre las clases y los métodos en el SDK de Vertex AI para Python, consulta la referencia del SDK de Vertex AI.

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 en un entorno aislado, puedes crear una instancia de Vertex AI Workbench. Luego, instala el SDK de Vertex AI y ejecuta tus secuencias de comandos de Python desde un notebook en tu instancia de Vertex AI Workbench. Para obtener más información, consulta Crea una instancia de Vertex AI Workbench.

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 inicializarlo 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?

• Obtén más información sobre el SDK de Vertex AI.