tensorflow - 'conda install tensorflow-gpu' 没有安装任何 CUDNN 库

Question

当我在新的 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 安装 tensorflow-gpu

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

Answer 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