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
revert-15207-remove_op_handle_lock_and_fix_var
qingqing01 6 years ago committed by GitHub
parent e29f10d315
commit a037378fdb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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]] +

Loading…
Cancel
Save