diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-06-26 11:16:17 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-26 11:19:40 -0700 |
| commit | d778904d942dcfcda8d21fe990cbb4e20102cfe9 (patch) | |
| tree | 1162767de64709d908306b03f6c7fb4236447cac /third_party | |
| parent | a4476d5bfefb6605a42d77da4598d5a8125e638d (diff) | |
Fixes to cuda_config.h.
PiperOrigin-RevId: 160168545
Diffstat (limited to 'third_party')
| -rw-r--r-- | third_party/toolchains/gpus/cuda/cuda/cuda_config.h | 13 |
1 files changed, 10 insertions, 3 deletions
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_ |