mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-09 00:30:53 +00:00
26 lines
1.4 KiB
Python
26 lines
1.4 KiB
Python
# -------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
# --------------------------------------------------------------------------
|
|
|
|
import os
|
|
import platform
|
|
import warnings
|
|
import onnxruntime.capi._ld_preload # noqa: F401
|
|
|
|
try:
|
|
from onnxruntime.capi.onnxruntime_pybind11_state import * # noqa
|
|
except ImportError as e:
|
|
warnings.warn("Cannot load onnxruntime.capi. Error: '{0}'.".format(str(e)))
|
|
|
|
# If on Windows, check if this import error is caused by the user not installing the 2019 VC Runtime
|
|
# The VC Redist installer usually puts the VC Runtime dlls in the System32 folder
|
|
# This may not always paint the true picture as anyone building from source using VS 2017 might hit this error
|
|
# because the machine might be missing the 2019 VC Runtime but it is not actually needed in that case and the
|
|
# import error might actually be due to some other reason.
|
|
# TODO: Add a guard against False Positive error message
|
|
# As a proxy for checking if the 2019 VC Runtime is installed,
|
|
# we look for a specific dll only shipped with the 2019 VC Runtime
|
|
if platform.system().lower() == 'windows' and not os.path.isfile('c:\\Windows\\System32\\vcruntime140_1.dll'):
|
|
warnings.warn("Unless you have built the wheel using VS 2017, "
|
|
"please install the 2019 Visual C++ runtime and then try again")
|