diff options
| author |  Guangda Lai <laigd@google.com> | 2018-09-06 13:34:35 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-06 13:43:58 -0700 |
| commit | d1e3df60b1202d67c8a1f8cc088c02a87481de26 (patch) | |
| tree | 337b7cc60eed9be44b281490bcb941b92adb0ee6 /tensorflow/tools/ci_build | |
| parent | 553ce1ff813fe7436ed3d6f194290e6703e2c179 (diff) | |
Set TF_CUDNN_VERSION to 7 in windows build. This doesn't change the version at the runtime since in configure.py it will strip the ".0" suffix, but it makes the things cleaner and less confusing.
PiperOrigin-RevId: 211860068
Diffstat (limited to 'tensorflow/tools/ci_build')
| -rw-r--r-- | tensorflow/tools/ci_build/windows/bazel/common_env.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/ci_build/windows/bazel/common_env.sh b/tensorflow/tools/ci_build/windows/bazel/common_env.sh index 333a89d3f5..c18f0d6e69 100644 --- a/tensorflow/tools/ci_build/windows/bazel/common_env.sh +++ b/tensorflow/tools/ci_build/windows/bazel/common_env.sh @@ -53,7 +53,7 @@ export PATH="/c/${PYTHON_BASE_PATH}/Scripts:$PATH" # Setting default values to CUDA related environment variables export TF_CUDA_VERSION=${TF_CUDA_VERSION:-9.0} -export TF_CUDNN_VERSION=${TF_CUDNN_VERSION:-7.0} +export TF_CUDNN_VERSION=${TF_CUDNN_VERSION:-7} export TF_CUDA_COMPUTE_CAPABILITIES=${TF_CUDA_COMPUTE_CAPABILITIES:-3.7} 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"} |