提交 e3949bf6 authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Merge pull request #4923 from genichyar/cudnn_fix

Fix cudnn when compiler-bindir specified in Windows. Fix issue #4545
...@@ -309,7 +309,7 @@ def dnn_available(): ...@@ -309,7 +309,7 @@ def dnn_available():
params.extend(['-L%s%s%s' % (path_wrapper, config.dnn.library_path, path_wrapper)]) params.extend(['-L%s%s%s' % (path_wrapper, config.dnn.library_path, path_wrapper)])
if config.nvcc.compiler_bindir: if config.nvcc.compiler_bindir:
params.extend(['--compiler-bindir', params.extend(['--compiler-bindir',
config.nvcc.compiler_bindir]) '%s%s%s' % (path_wrapper, config.nvcc.compiler_bindir, path_wrapper)])
params.extend([flag for flag in config.nvcc.flags.split(' ') if flag]) params.extend([flag for flag in config.nvcc.flags.split(' ') if flag])
# Do not run here the test program. It would run on the # Do not run here the test program. It would run on the
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论