如何将最新的cuDNN安装到conda?

时间:2019-03-20 08:46:32

标签: tensorflow conda cudnn

在conda中,conda的最新版本是:

cudnn                     7.3.1                cuda10.0_0    anaconda

但是我需要7.4.2 for tensorflow-gpu.1.13 如何在conda中安装cuDNN == 7.4.2?

6 个答案:

答案 0 :(得分:4)

最好的用途是使用 conda 环境安装 cuda-toolkit 和 CuDNN,以获得最佳兼容性。但在某些情况下,人们可能需要最新版本。此外,有时 cuda 软件包会以不同的时间表更新,例如暂时提供此答案,conda 提供 cudatoolkit-11.0 但不能同时提供 CuDNN-8.0。这发生在我的情况下。此问题有一个解决方法。

使用 conda 环境安装 conda-toolkit,并从 Nvidia CuDNN 页面下载最新匹配的 CuDNN 版本以安装 cuda-toolkit。使用 tar 并解压缩包并将 CuDNN 文件复制到您的 anaconda 环境中。

sudo cp cuda/include/cudnn*.h   /anaconda3/envs/<your environment here>/include
sudo cp cuda/lib64/libcudnn*    /anaconda3/envs/<your environment here>/lib
sudo chmod a+r /usr/local/cuda/include/cudnn*.h    /anaconda3/envs/<your environment here>/lib/libcudnn*

在给定的剪切“cuda”路径中代表解压缩的 CuDNN 文件夹。此解决方法已使用 tensorflow-2.4 & cudatoolkit-11.0 & CuDNN 8.0.4 进行测试

答案 1 :(得分:2)

  1. conda更新--force conda
  2. conda更新conda
  3. conda install -c anaconda cudnn
  4. conda清单cudnn

答案 2 :(得分:0)

我相信不可能用conda做到这一点。也许您可以尝试this

答案 3 :(得分:0)

  1. 您需要卸载cudnn:conda卸载cudnn。

  2. 卸载任何张量流依赖项:“ conda卸载tensorflow”

  3. 使用pip安装tensorflow:“ pip install tensorflow”

  4. 按照此处的说明安装CuDNN和Cuda ToolKit:https://www.tensorflow.org/install/gpu#linux_setup

  5. 使用PyCharm或Spyder使用tensorflow运行脚本

答案 4 :(得分:0)

这是我安装cudnn的方式。

1。您可以从NVIDIA下载所需版本的cudnn tar文件,然后将其解压缩。 然后,您可以看到包含cudnn文件的“ cuda”文件夹。

2。将cudnn文件复制并粘贴到conda envs lib并包含文件夹:

sudo cp cuda/include/cudnn*.h anaconda3/envs/"your_env_name"/include
sudo cp cuda/lib64/libcudnn* anaconda3/envs/"your_env_name"/lib

anaconda3是您的anaconda安装文件夹。

以我为例,

答案 5 :(得分:0)

您可以使用 conda-forge 安装

conda install -c conda-forge cudnn

https://anaconda.org/conda-forge/cudnn

它比 anaconda 频道更新 - 例如,截至今天,anaconda 上 cudnn 的最新版本仍然是 7.6.5,但在 conda-forge v8.2.0.53 上。 同样适用于 cudatoolkit 包。