【问题标题】:'conda install tensorflow-gpu' is not installing any CUDNN libraries'conda install tensorflow-gpu' 没有安装任何 CUDNN 库
【发布时间】:2020-12-23 04:33:03
【问题描述】:

当我在新的 venv 中运行 conda install tensorflow-gpu 时,conda 会给出它将安装的软件包列表,但没有我期望的 CUDNN 或 cudatoolkit 软件包,而其他人也有。我期待像这个问题一样的输出(Is it still necessary to install CUDA before using the conda tensorflow-gpu package? ) 但我得到了这个:

编辑:我可以通过运行这个命令来让它工作

conda create -n tf-gpu python=3.7 anaconda 后跟 conda install tensorflow-gpu

完成此操作后,它自行安装了所有 cudnn 库,并且我能够使用 GPU 运行 tensorflow。但是,由于某种原因,它正在安装 tensorflow 1.14,而我需要 2.x,所以有人知道这是为什么吗?

【问题讨论】:

  • This answer in a previous question会解决你的问题。

标签: tensorflow anaconda gpu


【解决方案1】:

conda install tensorflow-gpu 仅安装 GPU 版本的 Tensorflow,不支持 CUDA 和 cuDNN 等软件包。 按照步骤安装

conda install cudatoolkit=10.0.130
conda install cudnn=7.6.0=cuda10.0_0
conda install tensorflow-gpu

【讨论】:

  • 我在上面发布了一个编辑,我可以通过将 python=3.7 和 anaconda 添加到虚拟环境的创建中来使其工作。然后它按预期使用 tensorflow-gpu 命令安装了所有 cudnn 和 cudatoolkit 库。
猜你喜欢
  • 2020-10-02
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2017-02-10
  • 2017-12-08
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多