[Misc] Improve error message for incorrect pynvml (vllm-project#12809)

Signed-off-by: youkaichao <[email protected]>
Signed-off-by: Louis Ulmer <[email protected]>

Author: youkaichao

vllm/platforms/__init__.py

@@ -41,7 +41,11 @@ def cuda_platform_plugin() -> Optional[str]:
                 is_cuda = True
         finally:
             pynvml.nvmlShutdown()
-    except Exception:
+    except Exception as e:
+        if "nvml" not in e.__class__.__name__.lower():
+            # If the error is not related to NVML, re-raise it.
+            raise e
+
         # CUDA is supported on Jetson, but NVML may not be.
         import os