提交 70b53479 authored 作者: f0k's avatar f0k

Add CUDA include dir so cuda.h is found; also add CUDA library dir to mimic…

Add CUDA include dir so cuda.h is found; also add CUDA library dir to mimic configuration when compiling CUDA ops
上级 8834a330
......@@ -44,7 +44,9 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
"""
comp, run, out, err = gof.cmodule.GCC_compiler.try_flags(
["-l", "cudnn", "-I" + os.path.dirname(__file__)],
["-l", "cudnn", "-I" + os.path.dirname(__file__),
"-I" + os.path.join(theano.config.cuda.root, 'include'),
"-L" + os.path.join(theano.config.cuda.root, 'lib64')],
preambule=preambule, body=body,
try_run=True, output=True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论