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

Posted

技术标签:

【中文标题】如何将最新的 cuDNN 安装到 conda?【英文标题】:How to install latest cuDNN to conda? 【发布时间】:2019-08-10 21:43:50 【问题描述】:

在 conda 中最新版本的 conda 是:

cudnn                     7.3.1                cuda10.0_0    anaconda

但我需要 7.4.2 用于 tensorflow-gpu.1.13 conda中如何安装cuDNN==7.4.2?

【问题讨论】:

【参考方案1】:
    conda 更新 --force conda conda 更新 conda conda install -c anaconda cudnn conda list cudnn

【讨论】:

【参考方案2】:

    你需要卸载cudnn:conda uninstall cudnn。

    卸载任何 tensorflow 依赖项:“conda uninstall tensorflow”

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

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

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

【讨论】:

卸载后,我只需conda install tensorflow-gpu就可以得到相关的cuda版本【参考方案3】:

最好的用法是使用 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 进行测试

【讨论】:

【参考方案4】:

你可以用 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 包。

【讨论】:

【参考方案5】:

在提出问题时,使用 conda 无法做到这一点。这就是建议尝试this 的方式。然而,现在是可能的。关注其他答案

【讨论】:

嗯,现在可以anaconda.org/anaconda/cudnnconda install -c anaconda cudnn 【参考方案6】:

这就是我安装 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 安装文件夹。

就我而言,它有效。

【讨论】:

以上是关于如何将最新的 cuDNN 安装到 conda?的主要内容,如果未能解决你的问题,请参考以下文章

conda安装tensorflow总结

Ubuntu20.04安装CUDA+CUDNN+Conda+PyTorch

服务器从0搭建-anaconda3+cuda+cudnn+conda环境创建+修改conda源

Windows10 安装 Anaconda + CUDA + CUDNN (简明)

Windows10 安装 Anaconda + CUDA + CUDNN (简明图文教程)

linux服务器下命令行对anconda进行下载及其配置conda使用查看 cudnn& cuda