File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11#!/usr/bin/env python3
22
3- # Copyright 2018-2025 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3+ # Copyright 2018-2026 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
44#
55# Redistribution and use in source and binary forms, with or without
66# modification, are permitted provided that the following conditions
@@ -313,16 +313,13 @@ def check_gpus_compute_capability(min_capability):
313313 import importlib .util
314314
315315 if importlib .util .find_spec ("cuda" ) is not None :
316- import cuda .core .experimental as cuda_core_experimental
317-
318- devices = cuda_core_experimental .system .devices
316+ from cuda .core import Device
319317
318+ devices = Device .get_all_devices ()
320319 for device in devices :
321- compute_capability = (
322- device .compute_capability .major + device .compute_capability .minor / 10.0
323- )
324-
325- if compute_capability < min_capability :
320+ cc = device .compute_capability
321+ compute_capability_value = cc .major + cc .minor / 10.0
322+ if compute_capability_value < min_capability :
326323 return False
327324
328325 elif importlib .util .find_spec ("pycuda" ) is not None :
You can’t perform that action at this time.
0 commit comments