Tensorflow not running on GPU

前端 未结 3 743
無奈伤痛
無奈伤痛 2020-12-24 01:39

I have aldready spent a considerable of time digging around on stack overflow and else looking for the answer, but couldn\'t find anything

3条回答
  •  不思量自难忘°
    2020-12-24 02:13

    I was still having trouble getting GPU support even after correctly installing tensorflow-gpu via pip. My problem was that I had installed tensorflow 1.5, and CUDA 9.1 (the default version Nvidia directs you to), whereas the precompiled tensorflow 1.5 works with CUDA versions <= 9.0. Here is download page on nvidia's site to get the correct CUDA 9.0:

    https://developer.nvidia.com/cuda-90-download-archive

    Also make sure to update your cuDNN to a version compatible with CUDA 9.0 https://developer.nvidia.com/cudnn https://developer.nvidia.com/rdp/cudnn-download

提交回复
热议问题