提交 26d47b89 authored 作者: Frederic Bastien's avatar Frederic Bastien

Small update to error message

上级 e4dd0b3a
...@@ -329,7 +329,7 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) { ...@@ -329,7 +329,7 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
if v == -1 or v[0] < 4007: if v == -1 or v[0] < 4007:
# 4007 is the final release of cudnn v4 # 4007 is the final release of cudnn v4
dnn_available.avail = False dnn_available.avail = False
dnn_available.msg = "Version too old. Update to v5, was %d." % v[0] dnn_available.msg = "Version is too old. Update to v5, was %d." % v[0]
raise RuntimeError(dnn_available.msg) raise RuntimeError(dnn_available.msg)
else: else:
dnn_available.avail = comp dnn_available.avail = comp
......
...@@ -82,7 +82,7 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) { ...@@ -82,7 +82,7 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
def _dnn_check_version(): def _dnn_check_version():
v = version() v = version()
if v < 4007: if v < 4007:
return False, "Version too old. Update to v5, was %d." % v return False, "cuDNN version is too old. Update to v5, was %d." % v
return True, None return True, None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论