提交 18f2d661 authored 作者: Frederic Bastien's avatar Frederic Bastien

Don't call dnn_version if user disable dnn!

上级 4e6177e7
...@@ -576,6 +576,7 @@ def use(device, ...@@ -576,6 +576,7 @@ def use(device,
cudnn_version = "not available" cudnn_version = "not available"
warn = None warn = None
try: try:
if dnn_available():
(hdr_v, runtime_v) = dnn_version() (hdr_v, runtime_v) = dnn_version()
cudnn_version = runtime_v cudnn_version = runtime_v
# 4100 should not print warning with cudnn 4 final. # 4100 should not print warning with cudnn 4 final.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论