Skip to content

Instantly share code, notes, and snippets.

@leVirve
Last active March 22, 2017 06:44
Show Gist options
  • Save leVirve/04b37eada5a8b5dfba2c3c5a924f2487 to your computer and use it in GitHub Desktop.
Save leVirve/04b37eada5a8b5dfba2c3c5a924f2487 to your computer and use it in GitHub Desktop.
export JAVA_HOME="$(ls -d C:/Program\ Files/Java/jdk* | sort | tail -n 1)"
export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
export PATH=/c/ProgramData/Chocolatey/lib/bazel:"/c/Program Files/Git/cmd":$PATH
export BAZEL_VS="C:/Program Files (x86)/Microsoft Visual Studio 14.0"
export BAZEL_PYTHON="C:/Python27/python.exe"
#!/bin/bash
# Copyright 2016 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.
# ==============================================================================
#
# This script assumes the standard setup on tensorflow Jenkins windows machines.
# It is NOT guaranteed to work on any other machine. Use at your own risk!
#
# REQUIREMENTS:
# * All installed in standard locations:
# - JDK8, and JAVA_HOME set.
# - Microsoft Visual Studio 2015 Community Edition
# - Msys2
# - Anaconda3
# * Bazel windows executable copied as "bazel.exe" and included in PATH.
# Use a temporary directory with a short name.
export TMPDIR="E:/tmp"
mkdir -p "$TMPDIR"
# Set bash path
export BAZEL_SH="C:/tools/msys64/usr/bin/bash"
# Set Python path for ./configure
#export PYTHON_BIN_PATH="C:/Users/CVLab-Yao/AppData/Local/Programs/Python/Python35/python"
export PYTHON_BIN_PATH="C:/Python27/python"
# Set Python path for cc_configure.bzl
export BAZEL_PYTHON="C:/Python27/python"
# Set Visual Studio path
export BAZEL_VS="C:/Program Files (x86)/Microsoft Visual Studio 14.0"
# Add python into PATH, it's needed because gen_git_source.py uses
# '/usr/bin/env python' as a shebang
export PATH="/c/Python27:$PATH"
# Make sure we have pip in PATH
export PATH="/c/Python27/Scripts:$PATH"
# Add Cuda and Cudnn dll directories into PATH
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0/bin:$PATH"
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0/extras/CUPTI/libx64:$PATH"
export PATH="/c/tools/cuda/bin:$PATH"
# Set the common build options on Windows
export BUILD_OPTS='--cpu=x64_windows_msvc --host_cpu=x64_windows_msvc --copt=/w --verbose_failures --experimental_ui'
########################
### Settings for GPU ###
########################
export TF_NEED_CUDA=0 # TURN ONOFF GPU SUPPORT
export TF_CUDA_VERSION=8.0
export CUDA_TOOLKIT_PATH="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0"
export TF_CUDNN_VERSION=5
export CUDNN_INSTALL_PATH="C:/tools/cuda"
export TF_CUDA_COMPUTE_CAPABILITIES="3.5,5.2"
if [ -z "$TF_ENABLE_XLA" ]; then
export TF_ENABLE_XLA=0
fi
if [ -z "$CC_OPT_FLAGS" ]; then
export CC_OPT_FLAGS="-march=native"
fi
echo "" | ./configure
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment