Remove nvml-specific exception check in platform determination

vllm-project · Oct 29, 2024 · 400c3d1 · 400c3d1
1 parent da79e3e
commit 400c3d1
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/vllm/platforms/__init__.py b/vllm/platforms/__init__.py
@@ -27,9 +27,7 @@
             is_cuda = True
     finally:
         pynvml.nvmlShutdown()
-except ImportError:
-    pass
-except pynvml.NVMLError_LibraryNotFound:
+except Exception as e:
     # CUDA is supported on Jetson, but NVML is not.
     import os
 
@@ -40,8 +38,6 @@ def cuda_is_jetson() -> bool:
     if cuda_is_jetson():
         is_cuda = True
         is_cuda_jetson = True
-except Exception:
-    pass
 
 is_rocm = False