"conda install tensorflow-gpu"无法安装任何CUDNN库。

3
当我在新的虚拟环境中运行 conda install tensorflow-gpu 时,conda 给出了将要安装的软件包列表,但是没有 CUDNN 或 cudatoolkit 软件包,而这是我期望的,并且其他人已经得到。我期望的输出如此问题(Is it still necessary to install CUDA before using the conda tensorflow-gpu package?),但实际上我得到的是这样的:

The packages it is installing for me, I'm expecting CUDNN

编辑: 我通过运行以下命令成功地使其工作了

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

这样做后,它自己安装了所有的 cudnn 库,我能够使用 GPU 运行 tensorflow。然而,由于某种原因它正在安装 tensorflow 1.14 版本,而我需要 2.x 版本,有谁知道原因吗?


1
在之前的问题中,这个答案可以解决你的问题。 - Prabath
1个回答

0

conda install tensorflow-gpu 只安装 Tensorflow 的 GPU 版本,不包括 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库,就像预期的那样。 - A.lex

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接