提交 6b594491 authored 作者: abergeron's avatar abergeron 提交者: GitHub

Merge pull request #6011 from notoraptor/update-dnn-enabled-support

(small fix) update usage of `dnn.enabled`
......@@ -71,6 +71,10 @@ def init_dev(dev, name=None, preallocate=None):
avail = dnn.dnn_available(name)
if avail:
context.cudnn_handle = dnn._make_handle(context)
elif config.dnn.enabled == 'True':
raise RuntimeError(
"You enabled cuDNN, but we aren't able to use it: %s" %
dnn.dnn_available.msg)
if config.print_active_device:
if avail:
print("Using cuDNN version %d on context %s" % (dnn.version(), name),
......
......@@ -165,13 +165,7 @@ def dnn_present():
if dnn_present.avail:
dnn_present.avail, dnn_present.msg = _dnn_check_version()
if not dnn_present.avail:
raise RuntimeError(dnn_present.msg)
if config.dnn.enabled == "True":
if not dnn_present.avail:
raise RuntimeError(
"You enabled cuDNN, but we aren't able to use it: %s" %
dnn_present.msg)
return False
return dnn_present.avail
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论