Installer le SDK Vertex AI pour Python

Utilisez le SDK Vertex AI pour Python afin d'automatiser vos workflows de machine learning (ML). Cette rubrique explique comment installer le SDK Vertex AI pour Python. Pour en savoir plus sur le SDK Vertex AI, consultez les ressources suivantes :

L'installation du SDK Vertex AI pour Python comprend les étapes suivantes :

  1. Créer un environnement Python isolé
  2. Installer le package du SDK Vertex AI
  3. Initialiser le SDK Vertex AI

Créer un environnement Python isolé

Une bonne pratique Python consiste à installer le SDK Vertex AI dans un environnement Python isolé pour chaque projet. Cela permet d'éviter les conflits de dépendances, de versions et d'autorisations. Vous pouvez créer un environnement isolé pour utiliser la ligne de commande dans un shell ou pour utiliser un notebook.

Pour créer un environnement isolé lorsque vous utilisez la ligne de commande, activez un environnement venv. Une fois l'environnement venv activé, vous êtes prêt à installer le SDK Vertex AI et à exécuter vos scripts Python. Pour en savoir plus, consultez la page Isoler les dépendances à l'aide de venv et Configurer un environnement de développement Python.

Pour utiliser un notebook pour votre environnement isolé, créez une instance de notebook. Une fois votre instance de notebook créée, utilisez-la pour installer le SDK Vertex AI et exécuter vos scripts Python. Pour en savoir plus, consultez la page Créer une instance de notebooks gérés par l'utilisateur.

Installer ou mettre à jour le package du SDK Vertex AI

Pour installer ou mettre à jour le SDK Vertex AI, exécutez la commande suivante dans votre environnement virtuel :

pip install --upgrade google-cloud-aiplatform

Initialiser le SDK Vertex AI

Après avoir installé le SDK Vertex AI pour Python, vous devez l'initialiser avec vos détails Vertex AI et Google Cloud. Par exemple, lorsque vous initialisez le SDK, vous spécifiez des informations telles que le nom du projet, la région et le bucket Cloud Storage de préproduction. La méthode suivante est un exemple de méthode d'initialisation du SDK 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,
):

    from google.cloud import aiplatform

    aiplatform.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,
    )

Étapes suivantes