| # --- UORT bundle ------------------------------------------------------------ | |
| # Registers the directory of this module with the Windows loader before the | |
| # pybind11 extension is imported. Added by scripts\runtime\stage_python_bundle.ps1. | |
| try: | |
| from ._uort_dll_dirs import add_dll_dirs as _uort_add_dll_dirs | |
| _uort_add_dll_dirs() | |
| except Exception: # noqa: BLE001 - a plain package layout has no _uort_dll_dirs | |
| pass | |
| # ---------------------------------------------------------------------------- | |
| # ------------------------------------------------------------------------- | |
| # Copyright (c) Microsoft Corporation. All rights reserved. | |
| # Licensed under the MIT License. | |
| # -------------------------------------------------------------------------- | |
| """ | |
| Ensure that dependencies are available and then load the extension module. | |
| """ | |
| import os | |
| import platform | |
| import warnings | |
| from . import _ld_preload # noqa: F401 | |
| if platform.system() == "Windows": | |
| from . import version_info | |
| # 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, but it may also be found | |
| # in some other locations. | |
| # TODO, we may want to try to load the VC Runtime dlls instead of checking if the hardcoded file path | |
| # is valid, and raise ImportError if the load fails | |
| if version_info.vs2019 and platform.architecture()[0] == "64bit": | |
| system_root = os.getenv("SystemRoot") or "C:\\Windows" | |
| if not os.path.isfile(os.path.join(system_root, "System32", "vcruntime140_1.dll")): | |
| warnings.warn("Please install the 2019 Visual C++ runtime and then try again. " | |
| "If you've installed the runtime in a non-standard location " | |
| "(other than %SystemRoot%\\System32), " | |
| "make sure it can be found by setting the correct path.") | |
| from .onnxruntime_pybind11_state import * # noqa | |