diff options
Diffstat (limited to 'tensorflow/tools/ci_build/linux/gpu/run_cc_core.sh')
-rwxr-xr-x | tensorflow/tools/ci_build/linux/gpu/run_cc_core.sh | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/tools/ci_build/linux/gpu/run_cc_core.sh b/tensorflow/tools/ci_build/linux/gpu/run_cc_core.sh index df196f829c..ac83e90f76 100755 --- a/tensorflow/tools/ci_build/linux/gpu/run_cc_core.sh +++ b/tensorflow/tools/ci_build/linux/gpu/run_cc_core.sh @@ -28,6 +28,8 @@ echo "" export PYTHON_BIN_PATH=`which python3` export TF_NEED_CUDA=1 +export TF_CUDA_VERSION=8.0 +export TF_CUDNN_VERSION=6 export TF_CUDA_COMPUTE_CAPABILITIES=3.7 yes "" | $PYTHON_BIN_PATH configure.py |