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

Merge pull request #6038 from notoraptor/update-config-doc-for-dnn-paths

(small fix) Add doc for `dnn.include_path` and `dnn.library_path`.
...@@ -636,6 +636,18 @@ import theano and print the config variable, as in: ...@@ -636,6 +636,18 @@ import theano and print the config variable, as in:
If ``'no_check'``, assume present and the version between header and library match (so less compilation at context init)", If ``'no_check'``, assume present and the version between header and library match (so less compilation at context init)",
.. attribute:: config.dnn.include_path
Default: ``include`` sub-folder in CUDA root directory, or headers paths defined for the compiler.
Location of the cudnn header.
.. attribute:: config.dnn.library_path
Default: Library sub-folder (``lib64`` on Linux) in CUDA root directory, or libraries paths defined for the compiler.
Location of the cudnn library.
.. attribute:: config.conv.assert_shape .. attribute:: config.conv.assert_shape
If True, AbstractConv* ops will verify that user-provided If True, AbstractConv* ops will verify that user-provided
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论