如何将最新的 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?的主要内容,如果未能解决你的问题,请参考以下文章
Ubuntu20.04安装CUDA+CUDNN+Conda+PyTorch
服务器从0搭建-anaconda3+cuda+cudnn+conda环境创建+修改conda源
Windows10 安装 Anaconda + CUDA + CUDNN (简明)