diff --git a/vllm/platforms/__init__.py b/vllm/platforms/__init__.py index 9c98942b556..e4767a378f4 100644 --- a/vllm/platforms/__init__.py +++ b/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