| :: Copyright 2019 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. |
| :: ============================================================================= |
| echo on |
| |
| @REM |
| @REM Set Environment Variables |
| @REM |
| IF NOT DEFINED PYTHON_DIRECTORY ( |
| SET PYTHON_DIRECTORY=Python36 |
| ) |
| SET PY_EXE=C:\%PYTHON_DIRECTORY%\python.exe |
| SET PATH=%PATH%;C:\%PYTHON_DIRECTORY% |
| |
| @REM To have reproducible builds, these dependencies should be pinned always. |
| @REM Prefer pinning to the same version as in setup.py |
| @REM First, upgrade pypi wheels |
| %PY_EXE% -m pip install --upgrade setuptools pip wheel |
| @REM Now, install the deps, as listed in setup.py |
| %PY_EXE% -m pip install "absl-py ~= 0.10" |
| %PY_EXE% -m pip install "astunparse ~= 1.6.3" |
| %PY_EXE% -m pip install "flatbuffers ~= 1.12.0" |
| %PY_EXE% -m pip install "google_pasta ~= 0.2" |
| %PY_EXE% -m pip install "h5py ~= 3.1.0" |
| %PY_EXE% -m pip install "keras_preprocessing ~= 1.1.2" |
| %PY_EXE% -m pip install "numpy ~= 1.19.2" |
| %PY_EXE% -m pip install "opt_einsum ~= 3.3.0" |
| %PY_EXE% -m pip install "protobuf >= 3.9.2" |
| %PY_EXE% -m pip install "six ~= 1.15.0" |
| %PY_EXE% -m pip install "termcolor ~= 1.1.0" |
| %PY_EXE% -m pip install "typing_extensions ~= 3.7.4" |
| %PY_EXE% -m pip install "wheel ~= 0.35" |
| %PY_EXE% -m pip install "wrapt ~= 1.12.1" |
| @REM We need to pin gast dependency exactly |
| %PY_EXE% -m pip install "gast == 0.3.3" |
| @REM Finally, install tensorboard and estimator |
| @REM Note that here we want the latest version that matches (b/156523241) |
| %PY_EXE% -m pip install --upgrade --force-reinstall "tb-nightly ~= 2.4.0.a" |
| %PY_EXE% -m pip install --upgrade --force-reinstall "tensorflow_estimator ~= 2.3.0" |
| @REM Test dependencies |
| %PY_EXE% -m pip install "grpcio ~= 1.32.0" |
| %PY_EXE% -m pip install "portpicker ~= 1.3.1" |
| %PY_EXE% -m pip install "scipy ~= 1.5.2" |
| |
| :: Set cuda related environment variables. If we are not using CUDA, these are not used. |
| IF NOT DEFINED TF_CUDA_VERSION ( |
| SET TF_CUDA_VERSION=11.0 |
| ) |
| IF NOT DEFINED TF_CUDNN_VERSION ( |
| SET TF_CUDNN_VERSION=8 |
| ) |
| SET TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_50,sm_60,sm_70,sm_75,compute_80 |
| SET CUDA_TOOLKIT_PATH=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v%TF_CUDA_VERSION% |
| SET CUDNN_INSTALL_PATH=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v%TF_CUDA_VERSION% |
| SET PATH=%CUDA_TOOLKIT_PATH%\extras\CUPTI\libx64;%PATH% |
| SET PATH=%CUDA_TOOLKIT_PATH%\bin;%PATH% |
| SET PATH=%CUDNN_INSTALL_PATH%\bin;%PATH% |
| |
| @REM |
| @REM Setup Bazel |
| @REM |
| :: Download Bazel from github and make sure its found in PATH. |
| SET BAZEL_VERSION=3.1.0 |
| md C:\tools\bazel\ |
| wget -q https://github.com/bazelbuild/bazel/releases/download/%BAZEL_VERSION%/bazel-%BAZEL_VERSION%-windows-x86_64.exe -O C:/tools/bazel/bazel.exe |
| SET PATH=C:\tools\bazel;%PATH% |
| bazel version |