Tensorflow-GPU未将GPU与CUDA，CUDNN一起使用 [英] Tensorflow-GPU not using GPU with CUDA,CUDNN

问题描述

我想在GPU上使用Tensorflow.因此，我安装了所有必需的工具并按以下方式安装

I want to use Tensorflow on GPU. So I install all the needed tool and installed as below-

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

我使用deviseQuery示例测试了我的cuda，cudnn是否正常工作.但是Tensorflow没有使用GPU.然后我发现可能存在版本兼容性问题，因此我在下面的Tensorflow网站上使用conda环境检查安装了CudaToolkit，cudnn，并进行了版本兼容性检查.

I tested that my cuda,cudnn is working using deviseQuery example. But Tensorflow not used GPU. Then i find that version compatibility issue is possible so i innstalled CudaToolkit,cudnn using conda environment checking with version compatibility on Tensorflow website which is given below.

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

但是在此之后尝试Tensorflow-GPU尚未使用GPU.所以我现在在做什么?任何步骤或建议都需要.

But after this try Tensorflow-GPU not used GPU,yet. so what i am doing now? Any steps or suggestion require.

推荐答案

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

The installation engine has a problem for tensorflow-gpu 2.3 in Anaconda on Windows 10.

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

Workaround is to explicitly specify the correct tensorflow build:

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

这篇关于Tensorflow-GPU未将GPU与CUDA，CUDNN一起使用的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！