提交 8128e2d4 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix printing of cudnn version

上级 ce11f9a3
...@@ -60,7 +60,7 @@ def init_dev(dev, name=None): ...@@ -60,7 +60,7 @@ def init_dev(dev, name=None):
"updating Theano or downgrading cuDNN to " "updating Theano or downgrading cuDNN to "
"version 5.1.") "version 5.1.")
if config.print_active_device: if config.print_active_device:
print("Using cuDNN version %d" % (init_dev.dnn_version, name), print("Using cuDNN version %d" % init_dev.dnn_version,
file=sys.stderr) file=sys.stderr)
except Exception: except Exception:
pass pass
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论