Unverified 提交 e47f1c93 authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Merge pull request #6509 from nouiz/err_msg

Err msg et small opt in python code
...@@ -97,12 +97,18 @@ def _dnn_lib(): ...@@ -97,12 +97,18 @@ def _dnn_lib():
if lib_name: if lib_name:
break break
if lib_name is None: 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.base_path can help you. Current value "%s"' %
config.dnn.base_path)
else: else:
dnn_handle = ctypes.cdll.LoadLibrary(lib_name) dnn_handle = ctypes.cdll.LoadLibrary(lib_name)
if dnn_handle is None: 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.base_path can help you. Current value "%s"' %
config.dnn.base_path)
_dnn_lib.handle = dnn_handle _dnn_lib.handle = dnn_handle
cudnn = _dnn_lib.handle cudnn = _dnn_lib.handle
cudnn.cudnnCreate.argtypes = [ctypes.POINTER(ctypes.c_void_p)] cudnn.cudnnCreate.argtypes = [ctypes.POINTER(ctypes.c_void_p)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论