Remove nvml-specific exception check in platform determination

Signed-off-by: Conroy Cheers <conroy@corncheese.org>
vllm-project · Oct 29, 2024 · 51085cf · 51085cf
1 parent 36972a9
commit 51085cf
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:
     # 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