提交 bebbf915 authored 作者: Frederic Bastien's avatar Frederic Bastien

Raise the right type of error

上级 853bb088
...@@ -310,7 +310,7 @@ def version(raises=True): ...@@ -310,7 +310,7 @@ def version(raises=True):
""" """
if not dnn_present(): if not dnn_present():
if raises: if raises:
raise Exception( raise RuntimeError(
"We can't determine the cudnn version as it is not available", "We can't determine the cudnn version as it is not available",
dnn_available.msg) dnn_available.msg)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论