提交 286f2410 authored 作者: Frederic Bastien's avatar Frederic Bastien

Better error message.

上级 6b929c60
......@@ -97,12 +97,18 @@ def _dnn_lib():
if lib_name:
break
if lib_name is None:
raise RuntimeError('Could not find cudnn library (looked for v5* to v7*)')
raise RuntimeError(
'Could not find cudnn library (looked for v5* to v7*).'
' Check your cudnn installation. Maybe using the Theano'
' flag dnn.bin_path can help you. Current value "%s"' %
config.dnn.bin_path)
else:
dnn_handle = ctypes.cdll.LoadLibrary(lib_name)
if dnn_handle is None:
raise RuntimeError('Could not load cudnn library')
raise RuntimeError('Could not load cudnn library. Check your cudnn'
' installation. Maybe using the Theano'
' flag dnn.bin_path can help you. Current value "%s"' %
config.dnn.bin_path)
_dnn_lib.handle = dnn_handle
cudnn = _dnn_lib.handle
cudnn.cudnnCreate.argtypes = [ctypes.POINTER(ctypes.c_void_p)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论