如何在anaconda中使用默认的包以外的其他cudnn包

时间:2019-04-20 19:41:15

标签: windows tensorflow cudnn

我有一个anaconda安装程序,可在Windows 10上配置python 3.6,tensorflow 1.13。安装正常,除非当我遇到以下错误时尝试调用cudnn代码(例如conv2d和类似功能)时:“ E tensorflow / stream_executor /cuda/cuda_dnn.cc:324]已加载的运行时CuDNN库:7.3.1,但源代码编译为:7.4.1。如果CuDNN 7.0或更高版本,则CuDNN库的主要版本和次要版本必须匹配或具有更高的次要版本。如果使用二进制安装,请升级您的CuDNN库”。

Which TensorFlow and CUDA version combinations are compatible?。从这篇文章中,我尝试在桌面上安装cudnn 7.4.1,但看起来anaconda仅使用默认的cudnn 7.3.1,并且未引用路径变量所指向的那个。对于我如何强制conda使用cudnn 7.4.1的任何帮助,将不胜感激。

1 个答案:

答案 0 :(得分:0)

我不知道原始海报的问题是否已解决。万一有人遇到同样的问题(像我一样),则应首先尝试以下方法:

https://towardsdatascience.com/tensorflow-gpu-installation-made-easy-use-conda-instead-of-pip-52e5249374bc

基本上,建议您直接conda tensorflow-gpu并让conda处理cudnncuda版本。至少之后,如果您conda list,它将列出更新的(或更新的)cudnn版本。

但就我而言,此后,我仍然遇到相同的错误。所以我挖了一点,发现它来自我的其他包装之一。您可以在anaconda文件夹下搜索所有出现的cudnn64_7.dll。对我来说,mxnet-cu100包(即使使用最新版本)仍然使用旧的cudnn dll。我只是将较新的cudnn64_7.dll复制到该site-packages文件夹(我冒着一定的风险,但是在这个阶段,我没有什么可失去的),它解决了问题。