Fabric manager: support A100, H100, H200, B100 & B200 GPUs (#20)

gardenlinux · Mar 25, 2024 · 3e73618 · 3e73618
1 parent 8dc47b6
commit 3e73618
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/resources/install_fabricmanager.sh b/resources/install_fabricmanager.sh
@@ -6,7 +6,9 @@ source "$BIN_DIR"/set_env_vars.sh
 
 GPU_NAME=$("${NVIDIA_ROOT}"/bin/nvidia-smi -i 0 --query-gpu=name --format=csv,noheader)
 
-if [[ "$GPU_NAME" == *"A100"* ]]; then
+# Typical GPU name is something like "NVIDIA H100 80GB HBM3"
+# Fabric manager is required by the newer, bigger GPUs like A100, H100, etc. so we match those GPU types here
+if [[ "$GPU_NAME" =~ (A100|H100|H200|B100|B200) ]]; then
   OUTDIR=/out/nvidia-fabricmanager/$DRIVER_VERSION
   FABRICMANAGER_ARCHIVE="fabricmanager-linux-x86_64-$DRIVER_VERSION-archive"