wehub-resource-sync
72 lines
2.9 KiB
Python
72 lines
2.9 KiB
Python
# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
|
|
"""Platform-specific code for checking the integrity of the TensorFlow build."""
|
|
|
|
import os
|
|
|
|
MSVCP_DLL_NAMES = "msvcp_dll_names"
|
|
|
|
try:
|
|
from tensorflow.python.platform import build_info
|
|
except ImportError as exc:
|
|
raise ImportError(
|
|
"Could not import tensorflow. Do not import"
|
|
" tensorflow from its source directory; change directory to outside the"
|
|
" TensorFlow source tree, and relaunch your Python interpreter from"
|
|
" there."
|
|
) from exc
|
|
|
|
|
|
def preload_check():
|
|
"""Raises an exception if the environment is not correctly configured.
|
|
|
|
Raises:
|
|
ImportError: If the check detects that the environment is not correctly
|
|
configured, and attempting to load the TensorFlow runtime will fail.
|
|
"""
|
|
if os.name == "nt":
|
|
# Attempt to load any DLLs that the Python extension depends on before
|
|
# we load the Python extension, so that we can raise an actionable error
|
|
# message if they are not found.
|
|
if MSVCP_DLL_NAMES in build_info.build_info:
|
|
missing = []
|
|
for dll_name in build_info.build_info[MSVCP_DLL_NAMES].split(","):
|
|
found = False
|
|
for path_dir in os.environ.get("PATH", "").split(";"):
|
|
if os.path.isfile(os.path.join(path_dir, dll_name)):
|
|
found = True
|
|
break
|
|
if not found:
|
|
missing.append(dll_name)
|
|
if missing:
|
|
raise ImportError(
|
|
"Could not find the DLL(s) %r. TensorFlow requires that these DLLs "
|
|
"be installed in a directory that is named in your %%PATH%% "
|
|
"environment variable. You may install these DLLs by downloading "
|
|
'"Microsoft C++ Redistributable for Visual Studio 2015, 2017 and '
|
|
'2019" for your platform from this URL: '
|
|
"https://support.microsoft.com/help/2977003/the-latest-supported-visual-c-downloads"
|
|
% " or ".join(missing)
|
|
)
|
|
else:
|
|
# Load a library that performs CPU feature guard checking. Doing this here
|
|
# as a preload check makes it more likely that we detect any CPU feature
|
|
# incompatibilities before we trigger them (which would typically result in
|
|
# SIGILL).
|
|
from tensorflow.python.platform import _pywrap_cpu_feature_guard
|
|
|
|
_pywrap_cpu_feature_guard.InfoAboutUnusedCPUFeatures()
|