experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change default CUDA version on Windows to 10.1.

Browse Source 
This changes the release builds and presubmits.

PiperOrigin-RevId: 279121654
Change-Id: Iec7692d8ea5556dcff61ba67c29207a48aee7d65
This commit is contained in:
Christian Sigg 2019-11-07 10:54:24 -08:00 committed by TensorFlower Gardener
parent f7b58b559f
commit 9f36f88bdc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tools/ci_build/windows/bazel/common_env.sh
View File

@ -55,7 +55,7 @@ export PATH="/c/Program Files/Git/cmd:$PATH"
export PATH="/c/${PYTHON_BASE_PATH}/Scripts:$PATH"
# Setting default values to CUDA related environment variables
export TF_CUDA_VERSION=${TF_CUDA_VERSION:-10.0}
export TF_CUDA_VERSION=${TF_CUDA_VERSION:-10.1}
export TF_CUDNN_VERSION=${TF_CUDNN_VERSION:-7}
export TF_CUDA_COMPUTE_CAPABILITIES=${TF_CUDA_COMPUTE_CAPABILITIES:-6.0}
export CUDA_TOOLKIT_PATH=${CUDA_TOOLKIT_PATH:-"C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v${TF_CUDA_VERSION}"}