提交 89be3bcf authored 作者: Frederic Bastien's avatar Frederic Bastien

If cudnn isn't there and we use cuda back-end, print it.

上级 b7f0c300
...@@ -68,6 +68,7 @@ def init_dev(dev, name=None): ...@@ -68,6 +68,7 @@ def init_dev(dev, name=None):
warn = None warn = None
cudnn_version = "" cudnn_version = ""
if dev.startswith('cuda'): if dev.startswith('cuda'):
cudnn_version = " (CuDNN not available)"
try: try:
cudnn_version = dnn.version() cudnn_version = dnn.version()
# 4100 should not print warning with cudnn 4 final. # 4100 should not print warning with cudnn 4 final.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论