提交 fb5a99d0 authored 作者: ruslanagit's avatar ruslanagit

Use nvcc flags to compile with cuDNN

Fix for issue #4739
上级 dc49660f
...@@ -304,6 +304,8 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) { ...@@ -304,6 +304,8 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
if config.nvcc.compiler_bindir: if config.nvcc.compiler_bindir:
params.extend(['--compiler-bindir', params.extend(['--compiler-bindir',
config.nvcc.compiler_bindir]) config.nvcc.compiler_bindir])
[params.append(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
# default gpu, not the one selected by the user. If mixed # default gpu, not the one selected by the user. If mixed
# GPU are installed or if the GPUs are configured in # GPU are installed or if the GPUs are configured in
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论