提交 b7d23606 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Default to the cuda root for dnn.base_path if cudnn is there.

上级 a388cf41
......@@ -360,9 +360,18 @@ AddConfigVar('dnn.conv.precision',
in_c_key=False)
# We want to default to the cuda root if cudnn is installed there
def default_dnn_base_path():
root = get_cuda_root()
# The include doesn't change location between OS.
if root and os.path.exists(os.path.join(root, 'include', 'cudnn.h')):
return root
return ''
AddConfigVar('dnn.base_path',
"Install location of cuDNN.",
StrParam(''),
StrParam(default_dnn_base_path),
in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论