Fix test_cuda_device_order on some multi-GPU systems (#1590)

* Fix test_cuda_device_order

* Fix test

* Update cuda_bindings/tests/nvml/test_cuda.py

Co-authored-by: Ralf W. Grosse-Kunstleve <rwgkio@gmail.com>

---------

Co-authored-by: Ralf W. Grosse-Kunstleve <rwgkio@gmail.com>

Author: mdboom

cuda_bindings/tests/nvml/test_cuda.py

@@ -1,6 +1,8 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
+import os
+
 import cuda.bindings.driver as cuda
 from cuda.bindings import nvml
 
@@ -54,4 +56,12 @@ def test_cuda_device_order():
     cuda_devices = get_cuda_device_names()
     nvml_devices = get_nvml_device_names()
 
-    assert cuda_devices == nvml_devices, "CUDA and NVML device lists do not match"
+    if "CUDA_VISIBLE_DEVICES" not in os.environ:
+        # If that environment variable isn't set, the device lists should match exactly
+        assert cuda_devices == nvml_devices, "CUDA and NVML device lists do not match"
+    else:
+        # If the environment variable is set, there may possibly be fewer CUDA devices,
+        # and each of them should still be found in NVML devices.
+        assert len(cuda_devices) <= len(nvml_devices)
+        for cuda_device in cuda_devices:
+            assert cuda_device in nvml_devices, f"CUDA device {cuda_device} not found in NVML device list"