import onnxruntime as ort print(ort.get_device())