diff options
Diffstat (limited to 'tensorflow/tools/ci_build/windows/bazel/common_env.sh')
-rw-r--r-- | tensorflow/tools/ci_build/windows/bazel/common_env.sh | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/tools/ci_build/windows/bazel/common_env.sh b/tensorflow/tools/ci_build/windows/bazel/common_env.sh index 8a237e4e28..333a89d3f5 100644 --- a/tensorflow/tools/ci_build/windows/bazel/common_env.sh +++ b/tensorflow/tools/ci_build/windows/bazel/common_env.sh @@ -26,7 +26,8 @@ # * Bazel windows executable copied as "bazel.exe" and included in PATH. # Use a temporary directory with a short name. -export TMPDIR="C:/tmp" +export TMPDIR=${TMPDIR:-"C:/tmp"} +export TMPDIR=$(cygpath -m "$TMPDIR") mkdir -p "$TMPDIR" # Set bash path @@ -54,10 +55,10 @@ export PATH="/c/${PYTHON_BASE_PATH}/Scripts:$PATH" export TF_CUDA_VERSION=${TF_CUDA_VERSION:-9.0} export TF_CUDNN_VERSION=${TF_CUDNN_VERSION:-7.0} export TF_CUDA_COMPUTE_CAPABILITIES=${TF_CUDA_COMPUTE_CAPABILITIES:-3.7} -export CUDA_INSTALL_PATH=${CUDA_INSTALL_PATH:-"C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v${TF_CUDA_VERSION}"} +export CUDA_TOOLKIT_PATH=${CUDA_TOOLKIT_PATH:-"C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v${TF_CUDA_VERSION}"} export CUDNN_INSTALL_PATH=${CUDNN_INSTALL_PATH:-"C:/tools/cuda"} # Add Cuda and Cudnn dll directories into PATH -export PATH="$(cygpath -u "${CUDA_INSTALL_PATH}")/bin:$PATH" -export PATH="$(cygpath -u "${CUDA_INSTALL_PATH}")/extras/CUPTI/libx64:$PATH" +export PATH="$(cygpath -u "${CUDA_TOOLKIT_PATH}")/bin:$PATH" +export PATH="$(cygpath -u "${CUDA_TOOLKIT_PATH}")/extras/CUPTI/libx64:$PATH" export PATH="$(cygpath -u "${CUDNN_INSTALL_PATH}")/bin:$PATH" |