We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 449d1bc commit 0408efcCopy full SHA for 0408efc
vllm/platforms/__init__.py
@@ -41,7 +41,11 @@ def cuda_platform_plugin() -> Optional[str]:
41
is_cuda = True
42
finally:
43
pynvml.nvmlShutdown()
44
- except Exception:
+ except Exception as e:
45
+ if "nvml" not in e.__class__.__name__.lower():
46
+ # If the error is not related to NVML, re-raise it.
47
+ raise e
48
+
49
# CUDA is supported on Jetson, but NVML may not be.
50
import os
51
0 commit comments