Ative os tempos de execução predefinidos com GPUs

Para receber apoio técnico durante a pré-visualização, envie um email para vertex-notebooks-previews-external@google.com.

Este documento descreve como ativar um tempo de execução predefinido com GPUs para utilizadores do Colab Enterprise num projeto. Depois de ativar um tempo de execução predefinido com GPUs, os utilizadores podem mudar de um tempo de execução predefinido normal para um tempo de execução predefinido com GPUs clicando num botão no respetivo bloco de notas.

Este documento destina-se aos administradores do Colab Enterprise que querem ativar tempos de execução predefinidos com GPUs para outros utilizadores na respetiva organização. Pressupõe que tem conhecimentos sobre o seguinte:

  • Como gerir os tempos de execução e os modelos de tempo de execução do Colab Enterprise.
  • Como usar a gestão de identidade e de acesso (IAM) para controlar o acesso.

Vista geral

Para permitir que os utilizadores mudem para um tempo de execução predefinido com GPUs, um administrador (roles/aiplatform.colabEnterpriseAdmin) ou uma conta de utilizador com a autorização aiplatform.notebookRuntimeTemplates.create tem de criar primeiro um tempo de execução predefinido com GPUs.

Quando cria um tempo de execução predefinido com GPUs pela primeira vez, o Colab Enterprise cria um novo modelo de tempo de execução predefinido que inclui GPUs nas respetivas especificações. O tempo de execução predefinido original não é afetado e existe até expirar ou ser eliminado. Depois de criar o modelo de tempo de execução predefinido com GPUs, qualquer utilizador com a autorização aiplatform.notebookRuntimes.assign no projeto e a autorização aiplatform.notebookRuntimeTemplates.apply no modelo de tempo de execução pode criar e usar um tempo de execução predefinido com GPUs. Estas autorizações estão incluídas na função de utilizador do Colab Enterprise (roles/aiplatform.colabEnterpriseUser).

Especificações

O tempo de execução predefinido com GPUs tem especificações diferentes do tempo de execução predefinido original. Os tipos de máquinas, as GPUs e os tipos de discos de dados disponíveis variam consoante a região, pelo que algumas especificações podem ser diferentes do tempo de execução predefinido original.

A tabela seguinte descreve as especificações de um tempo de execução predefinido com GPUs com base na região do tempo de execução predefinido.

Descrição da região Especificações de tempo de execução predefinidas
Regiões que suportam GPUs L4
  • Tipo de máquina: g2-standard-4
  • Acelerador: 1 acelerador NVIDIA_L4
  • Disco de dados: 100 GB pd-balanced
Regiões que não suportam GPUs L4, mas suportam GPUs T4
  • Tipo de máquina: n1-standard-4
  • Acelerador: 1 acelerador NVIDIA_TESLA_T4
  • Disco de dados: 100 GB pd-standard
Regiões que não suportam GPUs L4 ou T4 Os tempos de execução predefinidos com GPUs não são suportados.

Disponibilidade do acelerador

O Colab Enterprise suporta tempos de execução predefinidos com os seguintes tipos de aceleradores:

  • L4
  • T4

Para saber mais sobre a disponibilidade regional destes aceleradores, consulte o artigo Usar aceleradores.

Antes de começar

  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.

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  3. Verify that billing is enabled for your Google Cloud project.

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

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the APIs

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

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  6. Verify that billing is enabled for your Google Cloud project.

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

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the APIs

  8. Funções necessárias

    Para garantir que a sua conta de utilizador tem as autorizações necessárias para ativar os tempos de execução predefinidos com GPUs no Colab Enterprise, peça ao seu administrador para conceder à sua conta de utilizador o papel de administrador do Colab Enterprise (roles/aiplatform.colabEnterpriseAdmin) IAM role on the project.

    For more information about granting roles, see Manage access to projects, folders, and organizations.

    Your administrator might also be able to give your user account the required permissions through custom roles or other predefined roles.

    Enable GPUs for default runtimes

    To enable GPUs for default runtimes, do the following:

    1. In the Google Cloud console, go to the Colab Enterprise My notebooks page.

      Go to My notebooks

    2. In the Region menu, select the region that contains your notebook.

    3. Click the notebook that you want to open. If you haven't created a notebook yet, create a notebook.

    4. In your notebook, click Connect.

    5. After Colab Enterprise connects to the default runtime, in the top right corner of your notebook, click the button to switch to a default runtime with GPUs. For example, if your notebook is in a region that supports L4 accelerators, click Switch to L4.

    Colab Enterprise creates a new default runtime that has GPUs, and then connects to the runtime. The ability to switch to a default runtime with GPUs is enabled for other users in the project.

    Turn off GPUs for default runtimes

    To turn off the ability to switch to a default runtime with GPUs, you must delete the runtime template named Default with GPU. See Delete a runtime template.

    Limitations

    • Default runtimes with GPUs are only available in regions that support specific accelerator availability. See Accelerator availability.

    • You must first connect a notebook to a default runtime to be able to switch to a default runtime with GPUs.

    What's next