[python]python onnxruntime gpu是否可用

import onnxruntime as ort
 
# 检查CUDA是否可用
def is_cuda_available():
    try:
        return ort.get_device() == 'GPU'
    except Exception:
        return False
 
# 检查ONNX Runtime是否支持CUDA
def is_onnxruntime_cuda_supported():
    return ort.get_device() == 'GPU'
 
# 检查ONNX Runtime是否使用CUDA
def is_onnxruntime_using_cuda():
    providers = ort.get_available_providers()
    return 'CUDA' in providers and 'CUDAExecutionProvider' in providers
 
# 测试代码
print(f"CUDA Available: {is_cuda_available()}")
print(f"ONNX Runtime CUDA Supported: {is_onnxruntime_cuda_supported()}")
print(f"ONNX Runtime Using CUDA: {is_onnxruntime_using_cuda()}")