# proto-file: https://cos.googlesource.com/cos/tools/+/refs/heads/master/src/pkg/gpuconfig/proto/gpu_driver_versions.proto
# proto-message: GPUDriverVerionInfoList

# Description:
# This file contains GPU driver version and label information.
# NVIDIA Release Feed:
# https://docs.nvidia.com/datacenter/tesla/drivers/releases.json

# Label Descriptions:
#   LATEST:  Points to the latest Production Branch (PB) / Long-Term Support Branch (LTSB) driver.
#   DEFAULT: Points to the Long-Term Support Branch (LTSB) otherwise, it will point to PB.
#   R[XXX]:  A label for pinning to a specific driver family (e.g., R570, R535). The major versions available will be from NVIDIA LTS or NVIDIA production branches


# --- Information on New Feature Branch (NFB) Drivers ---
# 1. This branch is intended for early adopters and testing and is not recommended for production workloads.
# 2. NFB drivers do not support CUDA Forward Compatibility. See NVIDIA's documentation for details:
# https://docs.nvidia.com/deploy/cuda-compatibility/#forward-compatibility
# 3. COS will remove it once a subsequent PB or LTSB driver is available.

# HARDWARE ENABLEMENT EXCEPTION:
# NVIDIA_RTX_PRO_6000 requires R575 as a minimum version.
# This New Feature Branch driver is included as LATEST/DEFAULT temporarily
# until a Production Branch is available for this hardware.
gpu_driver_version_info: [
    {
        gpu_device: {
            gpu_type: "NVIDIA_RTX_PRO_6000",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "DEFAULT",
                version: "575.57.08"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_B200",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "570.133.20"
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_H200",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "570.133.20"
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_V100",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_A100",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_A100_80GB",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_H100_80GB",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_P100",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_L4",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_P4",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_T4",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "OTHERS",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NO_GPU",
        },
        supported_driver_versions: [
            {
                label: "LATEST",
                version: "570.133.20"
            },
            {
                label: "DEFAULT",
                version: "535.247.01"
            },
            {
                version: "575.57.08" # New Feature Branch
            },
            {
                label: "R570",
                version: "570.133.20"
            },
            {
                version: "570.124.06"
            },
            {
                version: "570.86.15"
            },
            {
                label: "R565",
                version: "565.57.01"
            },
            {
                label: "R560",
                version: "560.35.03"
            },
            {
                label: "R550",
                version: "550.163.01"
            },
            {
                version: "550.144.03"
            },
            {
                version: "550.127.05"
            },
            {
                version: "550.90.12"
            },
            {
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.247.01"
            },
            {
                version: "535.230.02"
            },
            {
                version: "535.216.01"
            },
            {
                version: "535.183.06"
            },
            {
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            },
            {
                label: "R525",
                version: "525.147.05"
            },
            {
                version: "525.125.06"
            },
            {
                label: "R470",
                version: "470.256.02"
            },
            {
                version: "470.239.06"
            },
            {
                version: "470.223.02"
            },
            {
                version: "470.199.02"
            }
        ]
    }
]