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

Drop support for v2

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