From a037378fdb96773f44e0c12c14d2119b7e76996a Mon Sep 17 00:00:00 2001 From: qingqing01 Date: Wed, 9 Jan 2019 10:16:40 +0800 Subject: [PATCH] Fix error with cuDNN version less than 7.1. (#15219) Since conv_fusion_op is not exposed into Python, remote the env flag in __init__.py test=develop --- python/paddle/fluid/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py index f9f3807b156..2c17716500a 100644 --- a/python/paddle/fluid/__init__.py +++ b/python/paddle/fluid/__init__.py @@ -155,7 +155,7 @@ def __bootstrap__(): 'fraction_of_gpu_memory_to_use', 'cudnn_deterministic', 'enable_cublas_tensor_op_math', 'conv_workspace_size_limit', 'cudnn_exhaustive_search', 'memory_optimize_debug', 'selected_gpus', - 'cudnn_exhaustive_search_times', 'sync_nccl_allreduce' + 'sync_nccl_allreduce' ] core.init_gflags([sys.argv[0]] + -- GitLab