Activer les environnements d'exécution par défaut avec des GPU

Pour obtenir de l'aide pendant la version preview, envoyez un e-mail à vertex-notebooks-previews-external@google.com.

Ce document explique comment activer un environnement d'exécution par défaut avec des GPU pour les utilisateurs de Colab Enterprise dans un projet. Après avoir activé un environnement d'exécution par défaut avec des GPU, les utilisateurs peuvent passer d'un environnement d'exécution par défaut standard à un environnement d'exécution par défaut avec des GPU en cliquant sur un bouton dans leur notebook.

Ce document s'adresse aux administrateurs Colab Enterprise qui souhaitent activer les environnements d'exécution par défaut avec des GPU pour les autres utilisateurs de leur organisation. Il suppose que vous connaissez les éléments suivants:

  • Gérer les environnements d'exécution et les modèles d'environnement d'exécution Colab Enterprise
  • Utiliser Identity and Access Management (IAM) pour contrôler l'accès

Présentation

Pour permettre aux utilisateurs de passer à un environnement d'exécution par défaut avec des GPU, un administrateur (roles/aiplatform.colabEnterpriseAdmin) ou un compte utilisateur disposant de l'autorisation aiplatform.notebookRuntimeTemplates.create doit d'abord créer un environnement d'exécution par défaut avec des GPU.

La première fois que vous créez un environnement d'exécution par défaut avec des GPU, Colab Enterprise crée un modèle d'environnement d'exécution par défaut qui inclut des GPU dans ses spécifications. L'environnement d'exécution par défaut d'origine n'est pas affecté et existe jusqu'à son expiration ou sa suppression. Une fois le nouveau modèle d'environnement d'exécution par défaut avec GPU créé, tout utilisateur disposant de l'autorisation aiplatform.notebookRuntimes.assign sur le projet et de l'autorisation aiplatform.notebookRuntimeTemplates.apply sur le modèle d'environnement d'exécution peut créer et utiliser un environnement d'exécution par défaut avec GPU. Ces autorisations sont incluses dans le rôle "Utilisateur Colab Entreprise" (roles/aiplatform.colabEnterpriseUser).

Spécifications

L'environnement d'exécution par défaut avec GPU présente des spécifications différentes de celles de l'environnement d'exécution par défaut d'origine. Les types de machines, de GPU et de disques de données disponibles varient selon la région. Par conséquent, certaines spécifications peuvent être différentes de votre environnement d'exécution par défaut d'origine.

Le tableau suivant décrit les spécifications d'un environnement d'exécution par défaut avec des GPU en fonction de la région de l'environnement d'exécution par défaut.

Description de la région Spécifications d'environnement d'exécution par défaut
Régions compatibles avec les GPU L4
  • Type de machine : g2-standard-4
  • Accélérateur: 1 accélérateur NVIDIA_L4
  • Disque de données: 100 Go pd-balanced
Régions non compatibles avec les GPU L4, mais compatibles avec les GPU T4
  • Type de machine : n1-standard-4
  • Accélérateur: 1 accélérateur NVIDIA_TESLA_T4
  • Disque de données: 100 Go pd-standard
Régions non compatibles avec les GPU L4 ou T4 Les environnements d'exécution par défaut avec GPU ne sont pas compatibles.

Disponibilité de l'accélérateur

Colab Enterprise est compatible avec les environnements d'exécution par défaut avec les types d'accélérateurs suivants:

  • L4
  • T4

Pour en savoir plus sur la disponibilité régionale de ces accélérateurs, consultez la section Utiliser des accélérateurs.

Avant de commencer

  1. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.

  2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  3. Make sure that billing is enabled for your Google Cloud project.

  4. Enable the Vertex AI, Dataform, and Compute Engine APIs.

    Enable the APIs

    5.

  5. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  6. Make sure that billing is enabled for your Google Cloud project.

  7. Enable the Vertex AI, Dataform, and Compute Engine APIs.

    Enable the APIs

    8.

Rôles requis

Pour vous assurer que votre compte utilisateur dispose des autorisations nécessaires pour activer les environnements d'exécution par défaut avec des GPU dans Colab Enterprise, demandez à votre administrateur d'accorder à votre compte utilisateur le rôle IAM Administrateur Colab Enterprise (roles/aiplatform.colabEnterpriseAdmin) sur le projet.

Pour en savoir plus sur l'attribution de rôles, consultez Gérer l'accès aux projets, aux dossiers et aux organisations.

Votre administrateur peut également attribuer à votre compte utilisateur les autorisations requises via des rôles personnalisés ou d'autres rôles prédéfinis.

Activer les GPU pour les environnements d'exécution par défaut

Pour activer les GPU pour les environnements d'exécution par défaut, procédez comme suit:

  1. Dans la console Google Cloud, accédez à la page Mes notebooks de Colab Enterprise.

    Accéder à "Mes carnets"

  2. Dans le menu Région, sélectionnez la région qui contient votre notebook.

  3. Cliquez sur le notebook que vous souhaitez ouvrir. Si vous n'avez pas encore créé de notebook, créez-en un.

  4. Dans votre notebook, cliquez sur Connecter.

  5. Une fois que Colab Enterprise se connecte à l'environnement d'exécution par défaut, cliquez sur le bouton en haut à droite de votre notebook pour passer à un environnement d'exécution par défaut avec GPU. Par exemple, si votre ordinateur portable se trouve dans une région compatible avec les accélérateurs L4, cliquez sur Passer à L4.

Colab Enterprise crée un environnement d'exécution par défaut avec des GPU, puis s'y connecte. La possibilité de passer à un environnement d'exécution par défaut avec des GPU est activée pour les autres utilisateurs du projet.

Désactiver les GPU pour les environnements d'exécution par défaut

Pour désactiver la possibilité de passer à un environnement d'exécution par défaut avec GPU, vous devez supprimer le modèle d'environnement d'exécution nommé Default with GPU. Consultez Supprimer un modèle d'exécution.

Limites

  • Les environnements d'exécution par défaut avec GPU ne sont disponibles que dans les régions compatibles avec la disponibilité d'accélérateurs spécifiques. Consultez la section Disponibilité des accélérateurs.

  • Vous devez d'abord connecter un notebook à un environnement d'exécution par défaut pour pouvoir passer à un environnement d'exécution par défaut avec GPU.

Étape suivante