import onnxruntime

print(onnxruntime.get_device())
# ```
# 上面的代码给出的输出是'GPU'时，并不代表就成功了。
#
# 而要用下面的代码来验证：

# ```python
print('1',onnxruntime.get_available_providers())
# ort_session = onnxruntime.InferenceSession("weigth/best.onnx",providers=['CUDAExecutionProvider'])
# ort_session = onnxruntime.InferenceSession("weigth/best.onnx",providers=onnxruntime.get_available_providers())

ort_session = onnxruntime.InferenceSession("weigth/best.onnx")
print(ort_session.get_providers())

# ort_session = onnxruntime.InferenceSession("weigth/best.onnx")
# ort_session.set_providers(['CUDAExecutionProvider'], [{'device_id': 0}])
ort_session.set_providers(['CUDAExecutionProvider'],)
print(ort_session.get_providers())







import onnxruntime
print(onnxruntime.get_device())

print(3,onnxruntime.get_available_providers())
# ['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']

load_models(ort_session, queue, index,big_time)