提交 3c1728b8 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3664 from and-colonna/patch-1

Fixes cl.exe not being seen on Windows
...@@ -66,6 +66,8 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) { ...@@ -66,6 +66,8 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
params.append("-I" + config.dnn.include_path) params.append("-I" + config.dnn.include_path)
if config.dnn.library_path: if config.dnn.library_path:
params.append("-L" + 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 # 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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论