Tensorflow-GPU 不使用带有 CUDA、CUDNN 的 GPU

Tensorflow-GPU not using GPU with CUDA,CUDNN

我想在 GPU 上使用 Tensorflow。所以我安装了所有需要的工具并安装如下 -

  1. CUDA-11.2
  2. CUDNN-11.1
  3. Anaconda-2020.11
  4. Tensorflow-GPU-2.3.0

我使用 deviseQuery 示例测试了我的 cuda、cudnn 是否正常工作。 但是 Tensorflow 没有使用 GPU。然后我发现版本兼容性问题是可能的，所以我安装了 CudaToolkit，cudnn 使用 conda 环境检查 Tensorflow 网站上的版本兼容性，如下所示。

  1. CUDA-10.2.89
  2. CUDNN-7.6.5
  3. Tensorflow-GPU-2.3.0

但是在尝试了 Tensorflow-GPU 之后还没有使用 GPU。那我现在在做什么？任何步骤或建议都需要。

安装引擎在 Windows 10 上的 Anaconda 中 tensorflow-gpu 2.3 有问题。

解决方法是明确指定正确的 tensorflow 构建：

conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py38h1fcfbd6_0