Backport #1585 to 12.9.x

cuda_bindings/cuda/bindings/_test_helpers/arch_check.py

@@ -3,12 +3,31 @@
 
 
 from contextlib import contextmanager
+from functools import cache
 
 import pytest
 
 from cuda.bindings import nvml
 
 
+@cache
+def hardware_supports_nvml():
+    """
+    Tries to call the simplest NVML API possible to see if just the basics
+    works.  If not we are probably on one of the platforms where NVML is not
+    supported at all (e.g. Jetson Orin).
+    """
+    nvml.init_v2()
+    try:
+        nvml.system_get_driver_branch()
+    except (nvml.NotSupportedError, nvml.UnknownError):
+        return False
+    else:
+        return True
+    finally:
+        nvml.shutdown()
+
+
 @contextmanager
 def unsupported_before(device: int, expected_device_arch: nvml.DeviceArch | str | None):
     device_arch = nvml.device_get_architecture(device)

cuda_bindings/tests/nvml/__init__.py

@@ -1,2 +1,10 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
+
+
+import pytest
+
+from cuda.bindings._test_helpers.arch_check import hardware_supports_nvml
+
+if not hardware_supports_nvml():
+    pytest.skip("NVML not supported on this platform", allow_module_level=True)