提交 83e7717a authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Fix gpuarray cuDNN version check. (Fixes #5369.)

上级 9e6e051b
...@@ -135,7 +135,7 @@ def _dnn_check_version(): ...@@ -135,7 +135,7 @@ def _dnn_check_version():
if v < 5000: if v < 5000:
return False, "cuDNN version is too old. Update to v5, was %d." % v return False, "cuDNN version is too old. Update to v5, was %d." % v
# 5200 should not print warning with cudnn 5.1 final. # 5200 should not print warning with cudnn 5.1 final.
if version >= 5200: if v >= 5200:
warnings.warn("Your cuDNN version is more recent than " warnings.warn("Your cuDNN version is more recent than "
"Theano. If you encounter problems, try " "Theano. If you encounter problems, try "
"updating Theano or downgrading cuDNN to " "updating Theano or downgrading cuDNN to "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论