提交 5b10cc54 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix error message for cudnn library not found.

上级 d6903d00
......@@ -63,7 +63,7 @@ def _dnn_lib():
if lib_name:
break
if lib_name is None:
raise RuntimeError('Could not find cudnn library (maybe your are using a )')
raise RuntimeError('Could not find cudnn library (looked for v4 and v5[.1])')
_dnn_lib.handle = ctypes.cdll.LoadLibrary(lib_name)
cudnn = _dnn_lib.handle
cudnn.cudnnCreate.argtypes = [ctypes.POINTER(ctypes.c_void_p)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论