提交 f52b6e42 authored 作者: Andrea's avatar Andrea

Fixes cl.exe not being seen on Windows

comp, out and err evaluate to: False b"nvcc fatal : Cannot find compiler 'cl.exe' in PATH\r\n" b'' So I found it necessary to add the --compiler-bindir flag to the source, in order to actually use cudnn.
上级 dfb27303
......@@ -69,6 +69,8 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
params.append("-I" + config.dnn.include_path)
if config.dnn.library_path:
params.append("-L" + config.dnn.library_path)
if config.nvcc.compiler_bindir:
params.extend(['--compiler-bindir', config.nvcc.compiler_bindir])
# Do not run here the test program. It would run on the
# default gpu, not the one selected by the user. If mixed
# GPU are installed or if the GPUs are configured in
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论