diff options
Diffstat (limited to 'third_party/toolchains/gpus/cuda/cuda/cuda_config.h')
-rw-r--r-- | third_party/toolchains/gpus/cuda/cuda/cuda_config.h | 6 |
1 files changed, 3 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 f6662274cc..7cdaf144ad 100644 --- a/third_party/toolchains/gpus/cuda/cuda/cuda_config.h +++ b/third_party/toolchains/gpus/cuda/cuda/cuda_config.h @@ -19,9 +19,9 @@ limitations under the License. #define TF_CUDA_CAPABILITIES CudaVersion("3.0") -#define TF_CUDA_VERSION "8.0" -#define TF_CUDNN_VERSION "5" +#define TF_CUDA_VERSION "9.0" +#define TF_CUDNN_VERSION "7" -#define TF_CUDA_TOOLKIT_PATH "/usr/local/cuda-8.0" +#define TF_CUDA_TOOLKIT_PATH "/usr/local/cuda-9.0" #endif // CUDA_CUDA_CONFIG_H_ |