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

Question

我想在 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。那我现在在做什么？需要任何步骤或建议。

Answer 1

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

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

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