如何通过Azure REST API获取GPU虚拟机的显存大小？

直接通过Azure REST API获取的方案 #

你当前使用的2021-07-01版本Resource Skus List API确实无法返回GPU显存信息，但升级到2022-08-01及以后的API版本后，大部分GPU SKU的capabilities字段会新增GPUMemoryGB项，直接提供单卡显存大小。

调用地址（以2023-09-01版本为例）：

https://management.azure.com/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus?api-version=2023-09-01

升级后，对应SKU的capabilities会包含类似内容：

{
  "name": "GPUMemoryGB",
  "value": "16"
}

程序化映射GPU型号的替代方案 #

如果因限制无法升级API版本，可通过VM的family字段映射对应GPU型号，再关联显存参数：

• 提取API返回的family字段（如示例中的standardNVFamily）
• 维护一个预定义映射表，核心对应关系示例：
  • standardNVFamily → NVIDIA M60（16GB显存）
  • standardNCFamily → NVIDIA K80（12GB显存）
  • standardNCv2Family → NVIDIA P100（16GB显存）
  • standardNCv3Family → NVIDIA V100（16GB显存）
  • standardNDv2Family → NVIDIA V100（32GB显存）
• 程序化匹配后计算总显存（GPU数量 × 单卡显存）

你提供的API调用示例输出（2021-07-01版本）：

{
    "locations": [
        "eastus"
    ],
    "capabilities": [
        {
            "name": "MaxResourceVolumeMB",
            "value": "389120"
        },
        {
            "name": "OSVhdSizeMB",
            "value": "1047552"
        },
        {
            "name": "vCPUs",
            "value": "6"
        },
        ...
        {
            "name": "GPUs",
            "value": "1"
        },
        {
            "name": "vCPUsPerCore",
            "value": "1"
        }
    ],
    "locationInfo": [
        {
            "location": "eastus",
            "zones": [
                "useast-AZ01",
                "useast-AZ02"
            ]
        }
    ],
    "name": "Standard_NV6",
    "tier": "Standard",
    "size": "NV6",
    "family": "standardNVFamily"
}

内容的提问来源于stack exchange，提问作者joe_coolish