diff options
| -rwxr-xr-x | scripts/create_mdev.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/scripts/create_mdev.sh b/scripts/create_mdev.sh index 786e39b..2a5ad09 100755 --- a/scripts/create_mdev.sh +++ b/scripts/create_mdev.sh @@ -5,7 +5,7 @@ set -euo pipefail PCI_ADDR="${1:-}" -VGPU_TYPE="${2:-nvidia-nvidia-vgpu-turing}" +VGPU_TYPE="${2:-${VGPU_MDEV_TYPE:-nvidia-nvidia-vgpu-turing}}" # Supported vGPU types (matching the mdev type names the driver registers) VALID_TYPES=("nvidia-nvidia-vgpu-ampere" "nvidia-nvidia-vgpu-hopper" "nvidia-nvidia-vgpu-turing" "nvidia-nvidia-vgpu-turing-1g" "nvidia-nvidia-vgpu-turing-2g" "nvidia-nvidia-vgpu-pascal") |