提交 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) {
if v == -1 or v[0] < 4007:
# 4007 is the final release of cudnn v4
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)
else:
dnn_available.avail = comp
......
......@@ -82,7 +82,7 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
def _dnn_check_version():
v = version()
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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论