diff --git a/launcher/src/main.rs b/launcher/src/main.rs index c89a2432..c092d745 100644 --- a/launcher/src/main.rs +++ b/launcher/src/main.rs @@ -1655,6 +1655,7 @@ impl From<&str> for Gpu { "nvidia-h100-nvl" => Gpu::H100, "nvidia-h100" => Gpu::H100, "nvidia-a100-sxm4-80gb" => Gpu::A100, + "nvidia-a100-sxm4-40gb" => Gpu::A100, "nvidia-a100-80gb-pcie" => Gpu::A100, "nvidia-a100" => Gpu::A100, card => Gpu::Unknown(card.to_string()),