onnx_check.py 281 B

123456
  1. import onnxruntime
  2. print(onnxruntime.__version__)
  3. print(onnxruntime.get_device() ) # 如果得到的输出结果是GPU,所以按理说是找到了GPU的
  4. ort_session = onnxruntime.InferenceSession("fly.onnx",
  5. providers=['CUDAExecutionProvider'])
  6. print(ort_session.get_providers())