From d778904d942dcfcda8d21fe990cbb4e20102cfe9 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 26 Jun 2017 11:16:17 -0700 Subject: Fixes to cuda_config.h. PiperOrigin-RevId: 160168545 --- third_party/toolchains/gpus/cuda/cuda/cuda_config.h | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) (limited to 'third_party/toolchains') diff --git a/third_party/toolchains/gpus/cuda/cuda/cuda_config.h b/third_party/toolchains/gpus/cuda/cuda/cuda_config.h index dddee32193..f6662274cc 100644 --- a/third_party/toolchains/gpus/cuda/cuda/cuda_config.h +++ b/third_party/toolchains/gpus/cuda/cuda/cuda_config.h @@ -14,7 +14,14 @@ limitations under the License. ==============================================================================*/ // DO NOT EDIT: automatically generated file -#ifndef THIRD_PARTY_TENSORFLOW_OPENSOURCE_ONLY_TOOLCHAINS_GPUS_CUDA_CUDA_CONFIG_H_ -#define THIRD_PARTY_TENSORFLOW_OPENSOURCE_ONLY_TOOLCHAINS_GPUS_CUDA_CUDA_CONFIG_H_ +#ifndef CUDA_CUDA_CONFIG_H_ +#define CUDA_CUDA_CONFIG_H_ -#endif // THIRD_PARTY_TENSORFLOW_OPENSOURCE_ONLY_TOOLCHAINS_GPUS_CUDA_CUDA_CONFIG_H_ +#define TF_CUDA_CAPABILITIES CudaVersion("3.0") + +#define TF_CUDA_VERSION "8.0" +#define TF_CUDNN_VERSION "5" + +#define TF_CUDA_TOOLKIT_PATH "/usr/local/cuda-8.0" + +#endif // CUDA_CUDA_CONFIG_H_ -- cgit v1.2.3