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

Reference dnn.base_path

上级 286f2410
...@@ -100,15 +100,15 @@ def _dnn_lib(): ...@@ -100,15 +100,15 @@ def _dnn_lib():
raise RuntimeError( raise RuntimeError(
'Could not find cudnn library (looked for v5* to v7*).' 'Could not find cudnn library (looked for v5* to v7*).'
' Check your cudnn installation. Maybe using the Theano' ' Check your cudnn installation. Maybe using the Theano'
' flag dnn.bin_path can help you. Current value "%s"' % ' flag dnn.base_path can help you. Current value "%s"' %
config.dnn.bin_path) 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. Check your cudnn' raise RuntimeError('Could not load cudnn library. Check your cudnn'
' installation. Maybe using the Theano' ' installation. Maybe using the Theano'
' flag dnn.bin_path can help you. Current value "%s"' % ' flag dnn.base_path can help you. Current value "%s"' %
config.dnn.bin_path) 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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论