gke-gpu-driver-version

GKE GPU Driver Version

This tool helps you find the supported GPU driver version for a given GKE cluster version and GPU type. It automates the steps described in the official documentation for mapping GKE versions to corresponding COS and NVIDIA driver versions.

Prerequisites:

1. Install Rust
2. Clone COS tools into this directory
3. Install protobuf compiler

Usage

List supported GPU types

cargo run -- list-gpu-types --gke-version <CLUSTER_VERSION>

Get driver versions

cargo run -- get --gke-version <CLUSTER_VERSION> --gpu-type <GPU_TYPE>

Example

List available GPU types

cargo run -- list-gpu-types --gke-version 1.33.5-gke.1080000

Get driver versions

cargo run -- get --gke-version 1.33.5-gke.1080000 --gpu-type NVIDIA_H100_80GB

or

cargo run -- get --gke-version v1.33.5-gke.1080000 --gpu-type NVIDIA_H100_80GB