Tensorflow-GPU not using GPU with CUDA,CUDNN

不想你离开。 提交于 2021-01-28 11:15:11

问题


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

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

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


回答1:


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

Workaround is to explicitly specify the correct tensorflow build:

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


来源:https://stackoverflow.com/questions/65381073/tensorflow-gpu-not-using-gpu-with-cuda-cudnn

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!