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

Question

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

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

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

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

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

Answer 1

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

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

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