Cannot import Tensorflow for GPU on Windows 10

后端 未结 9 1805
被撕碎了的回忆
被撕碎了的回忆 2020-11-27 08:16

Trying to install the gpu version of Tensorflow on my machine, but I\'m getting this error when trying \'import tensorflow\'


ImportError

9条回答
  •  没有蜡笔的小新
    2020-11-27 08:36

    You need to have CUDA_HOME leading to \cuda and %CUDA_HOME%\bin appended to your PATH. Also, tensorflow as of now only supports cudNN 5.1, please make sure you have that.

提交回复
热议问题