提交 7ff6621e authored 作者: carriepl's avatar carriepl 提交者: Frederic

Drop support for v2

上级 3e4da0dc
...@@ -97,14 +97,14 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) { ...@@ -97,14 +97,14 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
dnn_available.msg = ( dnn_available.msg = (
"CuDNN v1 detected. This version is no longer " "CuDNN v1 detected. This version is no longer "
"supported by Theano. Update your CuDNN installation " "supported by Theano. Update your CuDNN installation "
"to a more recent version") "to v3 or more recent")
raise RuntimeError(dnn_available.msg) raise RuntimeError(dnn_available.msg)
if v == (20, 20): if v[0] < 3000:
dnn_available.avail = False dnn_available.avail = False
dnn_available.msg = ( dnn_available.msg = (
"You have installed a release candidate of CuDNN v2." "The detected CuDNN installation is older than CuDNN"
" This isn't supported anymore." " v3. This isn't supported by Theano anymore. Update"
" Update to CuDNN v2 final version.") " your CuDNN installation to v3 or more recent.")
raise RuntimeError(dnn_available.msg) raise RuntimeError(dnn_available.msg)
if 3000 <= v[0] < 3007: if 3000 <= v[0] < 3007:
# 3007 is the final release of cudnn v3 # 3007 is the final release of cudnn v3
......
...@@ -75,10 +75,10 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) { ...@@ -75,10 +75,10 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
def _dnn_check_version(): def _dnn_check_version():
v = version() v = version()
if v < 2000: if v < 3000:
return False, ( return False, (
"You have an old release of CuDNN (or a release candidate) " "You have an old release of CuDNN (or a release candidate) "
"that isn't supported. Please update to at least v2 final " "that isn't supported. Please update to at least v3 final "
"version.") "version.")
if 3000 <= v < 3007: if 3000 <= v < 3007:
return False, ( return False, (
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论