建立 AI 適用的 A4X MIG

本文說明如何建立使用 A4X 機型的代管執行個體群組 (MIG)。如要進一步瞭解這款加速器最佳化機型,請參閱「A4X」。

如要將多個虛擬機器 (VM) 視為單一實體管理,請建立 MIG。如要進一步瞭解 MIG,請參閱 Compute Engine 說明文件中的「代管執行個體群組」。

如要瞭解建立 VM 或叢集的其他方式,請參閱「部署選項總覽」。

限制

使用 A4X VM 建立 MIG 時,請考量機型和 MIG 的限制。

A4X VM 的限制

以下是機器專屬限制:

  • 使用這類機型的執行個體無法享有續用折扣或彈性承諾使用折扣。
  • 您只能在特定區域和可用區建立執行個體。
  • 您無法使用 Persistent Disk (區域性或可用區)。您只能使用 Google Cloud Hyperdisk。
  • 這類機器類型僅適用於 NVIDIA Grace 平台。
  • A4X 不支援變更機型。如要切換至或切換出這個機器類型,您必須建立新的執行個體。
  • 您無法在這個機器類型上執行 Windows 作業系統。如需支援的 Linux 作業系統清單,請參閱「GPU 執行個體支援的作業系統」。
  • 如果是 A4X 執行個體,當您使用 ethtool -S 監控 GPU 網路時,以 _phy 結尾的實體通訊埠計數器不會更新。對於使用 MRDMA 虛擬函式 (VF) 架構的執行個體,這是預期行為。詳情請參閱「MRDMA 函式和網路監控工具」。
  • A4X 執行個體不支援下列項目:
  • 2026 年 2 月 4 日前建立的 Hyperdisk ML 磁碟,無法連接至 A4X 機型。

MIG 的限制

使用 A4X 機器類型建立 MIG 時,須遵守下列限制:

  • 您必須使用取決於預留項目佈建模式。不支援其他佈建模式。

  • 您無法使用大小調整要求,將 A4X 執行個體新增至 MIG。您必須設定 MIG 的目標大小,才能新增執行個體。

事前準備

建立 MIG 前,請先完成下列步驟 (如已完成,請略過):

  1. 選擇使用模式:您選擇的使用模式會決定如何取得及使用 GPU 資源。詳情請參閱「選擇消耗選項」。
  2. 取得容量:取得容量的程序因消費選項而異。如要瞭解如何為所選用量方案取得運算資源,請參閱「運算資源總覽」。
  3. 驗證 Google API 服務代理權限:在 MIG 中建立 VM 時,Compute Engine 會使用專案中的 Google API 服務代理 (PROJECT_NUMBER@cloudservices.)。如果執行個體範本使用其他專案的資源,請確認這個服務代理具備必要權限:

必要的角色

如要取得建立 MIG 所需的權限,請要求管理員授予您專案的 Compute 執行個體管理員 (v1) (roles/compute.instanceAdmin.v1) IAM 角色。如要進一步瞭解如何授予角色,請參閱「管理專案、資料夾和組織的存取權」。

這個預先定義的角色具備建立 MIG 所需的權限。如要查看確切的必要權限,請展開「Required permissions」(必要權限) 部分:

所需權限

必須具備下列權限,才能建立 MIG:

  • 如要建立 MIG: compute.instanceGroupManagers.create 在專案中

您或許還可透過自訂角色或其他預先定義的角色取得這些權限。

A4X 基礎知識

A4X 叢集會依階層組織成模塊和子模塊,以利大規模、無阻塞的網路效能。預留容量和部署工作負載時,瞭解這個拓撲是關鍵。

A4X 執行個體
A4X 執行個體是單一 A4X 機型,連接 4 個 GPU。
子模塊
子模塊是 A4X 容量的基本單位。在 A4X 中,子區塊包含 18 個 A4X 執行個體 (72 個 GPU);這些執行個體會形成 NVLink 網域,並使用多節點 NVLink 系統連線。如要建立 A4X 子區塊,請套用指定 1x72 拓撲的工作負載政策。
封鎖
A4X 模塊由 25 個子模塊 (NVLink 網域) 組成,最多可包含 450 個 A4X 執行個體 (1,800 個 GPU)。子區塊會符合軌道排列,以便有效率地調整資源配置。每個子區塊都需要一個 MIG。因此,一個 A4X 區塊可以建立 25 個 MIG。

下表列出 A4X 執行個體支援的拓撲選項:

拓撲 (acceleratorTopology) GPU 數量 執行個體數量
1x72 72 18

總覽

建立 A4X 機型的 MIG 包含下列步驟:

  1. 建立虛擬私有雲網路
  2. 建立工作負載政策
  3. 建立執行個體範本
  4. 建立 MIG
  5. 安裝 GPU 驅動程式
  6. 選用:啟用 NVIDIA 多實體 GPU 模式

建立虛擬私有雲網路

如要設定 A4X 機器類型的網路,請為下列網路介面建立三個虛擬私有雲網路:

  • 2 個一般虛擬私有雲網路,用於 gVNIC 網路介面 (NIC)。NIC 接著會使用這些虛擬私有雲網路進行主機對主機通訊。
  • 建立多個 A4X 子區塊時,1 個虛擬私有雲網路會採用 RoCE 網路設定檔 ,用於 CX-7 NIC。 VM 執行個體的 RoCE 虛擬私有雲網路必須有 4 個子網路,每個 CX-7 NIC 各一個子網路。這些 NIC 採用 RoCE (基於融合乙太網路的 RDMA),提供高頻寬、低延遲的通訊,這對於向外擴充至多個 A4X 子模塊至關重要。

如要進一步瞭解 NIC 排列方式,請參閱「查看網路頻寬和 NIC 排列方式」。

您可以按照操作指南手動建立網路,也可以使用提供的指令碼自動建立。

操作指南

如要建立網路,請按照下列操作說明進行:

請將這些虛擬私有雲網路的最大傳輸單位 (MTU) 設為較大的值。如果是 A4X 機型,建議的 MTU 為 8896 位元組。如要查看其他 GPU 機型的建議 MTU 設定,請參閱「GPU 機型的 MTU 設定」。

指令碼

如要建立網路,請按照下列步驟操作。

請將這些虛擬私有雲網路的最大傳輸單位 (MTU) 設為較大的值。如果是 A4X 機型,建議的 MTU 為 8896 位元組。如要查看其他 GPU 機型的建議 MTU 設定,請參閱「GPU 機型的 MTU 設定」。

  1. 使用下列指令碼,為 gVNIC 建立一般虛擬私有雲網路。

    
    
        #!/bin/bash
    
        # Create regular VPC networks and subnets for the gVNICs
        for N in $(seq 0 1); do
          gcloud compute networks create GVNIC_NAME_PREFIX-net-$N \
            --subnet-mode=custom \
            --mtu=8896
    
          gcloud compute networks subnets create GVNIC_NAME_PREFIX-sub-$N \
            --network=GVNIC_NAME_PREFIX-net-$N \
            --region=REGION \
            --range=192.168.$N.0/24
    
          gcloud compute firewall-rules create GVNIC_NAME_PREFIX-internal-$N \
            --network=GVNIC_NAME_PREFIX-net-$N \
            --action=ALLOW \
            --rules=tcp:0-65535,udp:0-65535,icmp \
            --source-ranges=192.168.0.0/16
        done
    
        # Create SSH firewall rules
        gcloud compute firewall-rules create GVNIC_NAME_PREFIX-ssh \
          --network=GVNIC_NAME_PREFIX-net-0 \
          --action=ALLOW \
          --rules=tcp:22 \
          --source-ranges=IP_RANGE
    
        # Assumes that an external IP is only created for vNIC 0
        gcloud compute firewall-rules create GVNIC_NAME_PREFIX-allow-ping-net-0 \
          --network=GVNIC_NAME_PREFIX-net-0 \
          --action=ALLOW \
          --rules=icmp \
          --source-ranges=IP_RANGE
    
    
          
  2. 如需多個 A4X 子區塊,請使用下列指令碼,為 VM 執行個體 (roce) 建立 RoCE 虛擬私有雲網路,並為每個 A4X 執行個體上的四個 CX-7 NIC 建立子網路。

    
    
        #!/bin/bash
    
        # List and make sure network profiles exist in the machine type's zone
        gcloud compute network-profiles list --filter "location.name=ZONE"
    
        # Create network for RDMA NICs
        gcloud compute networks create RDMA_NAME_PREFIX-mrdma \
          --network-profile=ZONE-vpc-roce \
          --subnet-mode custom \
          --mtu=8896
    
        # Create subnets
        for N in $(seq 0 3); do
          gcloud compute networks subnets create RDMA_NAME_PREFIX-mrdma-sub-$N \
            --network=RDMA_NAME_PREFIX-mrdma \
            --region=REGION \
            --range=192.168.$((N+2)).0/24 # offset to avoid overlap with gVNICs
        done
    
    
          
  3. 更改下列內容:

    • GVNIC_NAME_PREFIX:自訂名稱前置字串,用於 gVNIC 的一般虛擬私有雲網路和子網路。
    • RDMA_NAME_PREFIX:用於 VM 執行個體 RoCE 虛擬私有雲網路 (roce) 和 CX-7 NIC 子網路的自訂名稱前置字串。
    • ZONE:指定要使用的機型所在的可用區,例如 us-central1-a。如要瞭解區域,請參閱「 各區域和可用區的 GPU 供應情形」。
    • REGION::要建立 子網路的區域。這個區域必須與指定的可用區相對應。舉例來說,如果可用區是 us-central1-a,則區域為 us-central1。
    • IP_RANGE:SSH 防火牆規則使用的 IP 範圍。
  4. 選用:如要確認虛擬私有雲網路資源已成功建立,請在 Google Cloud 控制台中檢查網路設定:
    1. 前往 Google Cloud 控制台的「VPC Networks」(虛擬私有雲網路) 頁面。

      前往「VPC networks」(虛擬私有雲網路)

    2. 在清單中搜尋您在上一個步驟建立的網路。
    3. 如要查看子網路、防火牆規則和其他網路設定,請按一下網路名稱。

建立執行個體範本

建立執行個體範本,指定 MIG 的 VM 屬性。

如要建立執行個體範本,請選取下列任一選項。

下列指令也會設定執行個體的存取權範圍。為簡化權限管理,Google 建議您將執行個體的存取權範圍設為 cloud-platform 存取權,然後使用 IAM 角色定義執行個體可存取的服務。詳情請參閱「範圍最佳做法」。

gcloud

如要建立區域性執行個體範本,請使用 gcloud compute instance-templates create 指令。

A4X 執行個體支援下列預留類型:

  • 預留未來容量區塊
  • 日曆模式的未來預留項目

如要使用這類預留項目,執行個體必須採用取決於預留項目的佈建模式。舉例來說,請使用下列建立參數。

gcloud compute instance-templates create INSTANCE_TEMPLATE_NAME \
    --machine-type=a4x-highgpu-4g \
    --image-family=IMAGE_FAMILY \
    --image-project=IMAGE_PROJECT \
    --instance-template-region=REGION \
    --boot-disk-type=hyperdisk-balanced \
    --boot-disk-size=DISK_SIZE \
    --scopes=cloud-platform \
    --network-interface=nic-type=GVNIC,network=GVNIC_NAME_PREFIX-net-0,subnet=GVNIC_NAME_PREFIX-sub-0 \
    --network-interface=nic-type=GVNIC,network=GVNIC_NAME_PREFIX-net-1,subnet=GVNIC_NAME_PREFIX-sub-1,no-address 
    --network-interface=nic-type=MRDMA,network=RDMA_NAME_PREFIX-mrdma,subnet=RDMA_NAME_PREFIX-mrdma-sub-0,no-address \
    --network-interface=nic-type=MRDMA,network=RDMA_NAME_PREFIX-mrdma,subnet=RDMA_NAME_PREFIX-mrdma-sub-1,no-address \
    --network-interface=nic-type=MRDMA,network=RDMA_NAME_PREFIX-mrdma,subnet=RDMA_NAME_PREFIX-mrdma-sub-2,no-address \
    --network-interface=nic-type=MRDMA,network=RDMA_NAME_PREFIX-mrdma,subnet=RDMA_NAME_PREFIX-mrdma-sub-3,no-address \
    --reservation-affinity=specific \
    --reservation=RESERVATION \
    --provisioning-model=RESERVATION_BOUND \
    --instance-termination-action=DELETE \
    --maintenance-policy=TERMINATE \
    --restart-on-failure

更改下列內容:

  • INSTANCE_TEMPLATE_NAME:執行個體範本的名稱。
  • IMAGE_FAMILY:要使用的 OS 映像檔系列。如需所有支援作業系統的清單,請參閱「支援的作業系統」。
  • IMAGE_PROJECT:OS 映像檔的專案 ID。
  • REGION:要建立執行個體範本的區域。指定要使用的機器類型所在的區域。如要瞭解區域,請參閱「依區域和可用區劃分的 GPU 可用性」。
  • DISK_SIZE:開機磁碟大小 (GB)。詳情請依磁碟類型參閱「Google Cloud Hyperdisk 大小限制」或「永久磁碟大小限制」。
  • GVNIC_NAME_PREFIX:建立使用 gVNIC NIC 的標準虛擬私有雲網路和子網路時,指定的名稱前置字串。
  • RDMA_NAME_PREFIX:建立使用 RDMA NIC 的虛擬私有雲網路和子網路時,指定的名稱前置字串。
  • RESERVATION:預留項目名稱、區塊或預留項目中的子區塊。如要取得預留項目名稱或可用區塊,請參閱「查看預留容量」。請根據執行個體位置需求,選擇下列其中一項:
    • 如要在任何單一區塊上建立 A4X 執行個體,請按照下列步驟操作:
          projects/RESERVATION_OWNER_PROJECT_ID/reservations/RESERVATION_NAME
          
    • 如要在特定區塊建立 A4X 廣告空間,請按照下列步驟操作:
          projects/RESERVATION_OWNER_PROJECT_ID/reservations/RESERVATION_NAME/reservationBlocks/RESERVATION_BLOCK_NAME
          
    • 如要在特定子區塊中建立 A4X 執行個體,請按照下列步驟操作:
          projects/RESERVATION_OWNER_PROJECT_ID/reservations/RESERVATION_NAME/reservationBlocks/RESERVATION_BLOCK_NAME/reservationSubBlocks/RESERVATION_SUBBLOCK_NAME
          

REST

如要建立區域執行個體範本,請向 regionInstanceTemplates.insert 方法發出 POST 要求。

A4X 執行個體支援下列預留類型:

  • 預留未來容量區塊
  • 日曆模式的未來預留項目

如要使用這類預留項目,執行個體必須採用取決於預留項目的佈建模式。舉例來說,請使用下列建立參數。

POST https://compute.googleapis.com/compute/v1/projects/PROJECT_ID/regions/REGION/instanceTemplates
{
  "name": "INSTANCE_TEMPLATE_NAME",
  "properties": {
    "machineType": "a4x-highgpu-4g",
    "disks": [
      {
        "boot": true,
        "initializeParams": {
          "diskSizeGb": "DISK_SIZE",
          "diskType": "hyperdisk-balanced",
          "sourceImage": "projects/IMAGE_PROJECT/global/images/family/IMAGE_FAMILY"
        },
        "mode": "READ_WRITE",
        "type": "PERSISTENT"
      }
    ],
    "serviceAccounts": [
      {
        "email": "default",
        "scopes": [
          "https://www.googleapis.com/auth/cloud-platform"
        ]
      }
    ],
    "networkInterfaces": [
      {
        "accessConfigs": [
          {
            "name": "external-nat",
            "type": "ONE_TO_ONE_NAT"
          }
        ],
        "network": "projects/NETWORK_PROJECT_ID/global/networks/GVNIC_NAME_PREFIX-net-0",
        "nicType": "GVNIC",
        "subnetwork": "projects/NETWORK_PROJECT_ID/region/REGION/subnetworks/GVNIC_NAME_PREFIX-sub-0"
      },
      {
        "network": "projects/NETWORK_PROJECT_ID/global/networks/GVNIC_NAME_PREFIX-net-1",
        "nicType": "GVNIC",
        "subnetwork": "projects/NETWORK_PROJECT_ID/region/REGION/subnetworks/GVNIC_NAME_PREFIX-sub-1"
      },
      {
        "network": "projects/NETWORK_PROJECT_ID/global/networks/RDMA_NAME_PREFIX-mrdma",
        "nicType": "MRDMA",
        "subnetwork": "projects/NETWORK_PROJECT_ID/region/REGION/subnetworks/RDMA_NAME_PREFIX-mrdma-sub-0"
      },
      {
        "network": "projects/NETWORK_PROJECT_ID/global/networks/RDMA_NAME_PREFIX-mrdma",
        "nicType": "MRDMA",
        "subnetwork": "projects/NETWORK_PROJECT_ID/region/REGION/subnetworks/RDMA_NAME_PREFIX-mrdma-sub-1"
      },
      {
        "network": "projects/NETWORK_PROJECT_ID/global/networks/RDMA_NAME_PREFIX-mrdma",
        "nicType": "MRDMA",
        "subnetwork": "projects/NETWORK_PROJECT_ID/region/REGION/subnetworks/RDMA_NAME_PREFIX-mrdma-sub-2"
      },
      {
        "network": "projects/NETWORK_PROJECT_ID/global/networks/RDMA_NAME_PREFIX-mrdma",
        "nicType": "MRDMA",
        "subnetwork": "projects/NETWORK_PROJECT_ID/region/REGION/subnetworks/RDMA_NAME_PREFIX-mrdma-sub-3"
      }
    ],
    "reservationAffinity": {
        "consumeReservationType": "SPECIFIC_RESERVATION",
        "key": "compute.googleapis.com/reservation-name",
        "values": [
          "RESERVATION"
        ]
      },
    "scheduling": {
        "provisioningModel": "RESERVATION_BOUND",
        "instanceTerminationAction": "DELETE",
        "onHostMaintenance": "TERMINATE",
        "automaticRestart": true
      }
  }
}

更改下列內容:

  • INSTANCE_TEMPLATE_NAME:執行個體範本的名稱。
  • IMAGE_FAMILY:要使用的 OS 映像檔系列。如需所有支援作業系統的清單,請參閱「支援的作業系統」。
  • IMAGE_PROJECT:OS 映像檔的專案 ID。
  • REGION:要建立執行個體範本的區域。指定提供您要使用機器類型的區域。如要瞭解區域,請參閱「各區域和可用區的 GPU 供應情形」。
  • DISK_SIZE:開機磁碟大小 (GB)。詳情請依磁碟類型參閱「Google Cloud Hyperdisk 大小限制」或「永久磁碟大小限制」。
  • NETWORK_PROJECT_ID:網路的專案 ID。
  • GVNIC_NAME_PREFIX:建立使用 gVNIC NIC 的標準虛擬私有雲網路和子網路時,指定的名稱前置字串。
  • REGION:子網路的區域。
  • RDMA_NAME_PREFIX:建立使用 RDMA NIC 的虛擬私有雲網路和子網路時,指定的名稱前置字串。
  • RESERVATION:預留項目名稱、區塊或預留項目中的子區塊。如要取得預留項目名稱或可用區塊,請參閱「查看預留容量」。請根據執行個體位置需求,選擇下列其中一項:
    • 如要在任何單一區塊上建立 A4X 執行個體,請按照下列步驟操作:
          projects/RESERVATION_OWNER_PROJECT_ID/reservations/RESERVATION_NAME
          
    • 如要在特定區塊建立 A4X 廣告空間,請按照下列步驟操作:
          projects/RESERVATION_OWNER_PROJECT_ID/reservations/RESERVATION_NAME/reservationBlocks/RESERVATION_BLOCK_NAME
          
    • 如要在特定子區塊中建立 A4X 執行個體,請按照下列步驟操作:
          projects/RESERVATION_OWNER_PROJECT_ID/reservations/RESERVATION_NAME/reservationBlocks/RESERVATION_BLOCK_NAME/reservationSubBlocks/RESERVATION_SUBBLOCK_NAME
          

建立 MIG

使用 A4X 機器類型時,您必須指定目標大小,才能建立 MIG。

如要取得 1x72 的 GPU 拓撲,請建立包含 18 個 A4X 執行個體的 MIG。建立 MIG 時,請套用指定 acceleratorTopology 欄位的工作負載政策。套用政策後,Compute Engine 會在一個子區塊中建立所有 18 個 A4X 執行個體,以便使用 NVLink 網域。如果子區塊沒有足夠容量容納所有 18 個執行個體,則無法立即建立的代管執行個體會維持在 CREATING 狀態,直到有可用容量為止。這些代管執行個體代表 MIG 在有容量時建立的運算執行個體。

如果套用 acceleratorTopology 欄位設為 1x72 的工作負載政策,則無法在 MIG 中建立超過 18 個 A4X 執行個體。如果指定超過 18 個執行個體,MIG 建立作業就會失敗。如要建立多個 NVLink 網域,請為每個網域建立獨立的 MIG,並對每個 MIG 套用相同的工作負載政策。

如要建立 MIG,請選取下列任一選項:

gcloud

如要建立具有指定目標大小的 MIG,請使用 instance-groups managed create 指令。

建立可用區或區域 MIG 的步驟如下:

  • 如要建立可用區 MIG,請使用下列指令:
    gcloud compute instance-groups managed create MIG_NAME \
      --template=INSTANCE_TEMPLATE_URL \
      --size=TARGET_SIZE \
      --workload-policy=WORKLOAD_POLICY_URL \
      --zone=ZONE
    
  • 如要建立區域性 MIG,請使用下列指令:
    gcloud compute instance-groups managed create MIG_NAME \
        --template=INSTANCE_TEMPLATE_URL \
        --size=TARGET_SIZE \
        --workload-policy=WORKLOAD_POLICY_URL \
        --region=REGION
    
更改下列內容:
  • MIG_NAME:MIG 的名稱。
  • INSTANCE_TEMPLATE_URL:您要用來在 MIG 中建立執行個體的執行個體範本網址。網址可以包含執行個體範本的 ID 或名稱。請指定下列其中一個值:
    • 區域性執行個體範本:projects/PROJECT_ID/regions/REGION/instanceTemplates/INSTANCE_TEMPLATE_ID
    • 全域執行個體範本:INSTANCE_TEMPLATE_ID
  • TARGET_SIZE:MIG 中要有的執行個體數量。針對加速器拓撲 1x72,將目標大小設為 18。
  • WORKLOAD_POLICY_URL:工作負載政策的網址,例如 projects/example-project/regions/us-central1/resourcePolicies/example-workload-policy。
  • ZONE:要建立 MIG 的可用區。指定工作負載政策區域內的可用區。
  • REGION:要建立 MIG 的區域。請指定與工作負載政策相同的區域。如果是區域性 MIG,您可以使用 --zones 旗標指定該區域中的可用區,而非指定區域。

REST

如要建立指定目標大小的 MIG,請提出 POST 要求。

建立可用區或區域 MIG 的步驟如下:

  • 如要建立可用區 MIG,請向 instanceGroupManagers.insert 方法發出 POST 要求。
    POST https://compute.googleapis.com/compute/v1/projects/PROJECT_ID/zones/ZONE/instanceGroupManagers
    {
      "versions": [
        {
          "instanceTemplate": "INSTANCE_TEMPLATE_URL"
        }
      ],
      "name": "MIG_NAME",
      "targetSize": TARGET_SIZE,
      "resourcePolicies": {
        "workloadPolicy": "WORKLOAD_POLICY_URL"
      }
    }
    
  • 如要建立區域性 MIG,請向 regionInstanceGroupManagers.insert 方法發出 POST 要求。
      POST https://compute.googleapis.com/compute/v1/projects/PROJECT_ID/regions/REGION/instanceGroupManagers
      {
        "name": "MIG_NAME",
        "instanceTemplate": "INSTANCE_TEMPLATE_URL",
        "targetSize": TARGET_SIZE,
        "resourcePolicies": {
          "workloadPolicy": "WORKLOAD_POLICY_URL"
        }
      }
    
更改下列內容:
  • PROJECT_ID:專案 ID。
  • ZONE:要建立 MIG 的可用區。指定工作負載政策區域內的可用區。
  • REGION:要建立 MIG 的區域。指定與工作負載政策相同的區域。
  • INSTANCE_TEMPLATE_URL:您要用來在 MIG 中建立執行個體的執行個體範本網址。網址可以包含執行個體範本的 ID 或名稱。請指定下列其中一個值:
    • 區域性執行個體範本:projects/PROJECT_ID/regions/REGION/instanceTemplates/INSTANCE_TEMPLATE_ID
    • 全域執行個體範本:INSTANCE_TEMPLATE_ID
  • MIG_NAME:MIG 的名稱。
  • TARGET_SIZE:MIG 中要有的執行個體數量。針對加速器拓撲 1x72,將目標大小設為 18。
  • WORKLOAD_POLICY_URL:工作負載政策的網址,例如 projects/example-project/regions/us-central1/resourcePolicies/example-workload-policy。

安裝 GPU 驅動程式

建立執行個體後,必須安裝正確的 GPU 驅動程式,執行個體才能使用 GPU。 如果您未指定包含 GPU 驅動程式的 OS 映像檔,請參閱 Compute Engine 說明文件中的「安裝 GPU 驅動程式」。

選用:啟用 NVIDIA 多實體 GPU 模式

多實體 GPU (MIG) 模式是可為支援的 NVIDIA GPU 啟用的功能。建立執行個體後,您可以在連接至機器的單一 GPU 上啟用 MIG 模式。啟用 MIG 模式後,單一 GPU 會分割為最多七個獨立 GPU 執行個體。每個執行個體會同時執行,且各自擁有記憶體、快取和串流多處理器。接著,您可以在這些 GPU 執行個體上平行執行不同的工作負載。 如要進一步瞭解如何使用 MIG 模式,請參閱 NVIDIA 說明文件中的多重執行個體 GPU (MIG) 使用手冊。

後續步驟