踩了很多坑,浪费了一下午时间,在此记录一下。
pip install tensorflow-gpu
然后下载安装cuda8.0
注意:只能用8.0
我用9.0下载了好久,程序提示我装8.0
然后:下载cudnn
要注册才能下载,这又是一个坑,选定cudnn6.0版本
其中,cudnn的安装方法是将cudnn下面的三个文件夹(bin,include, lib)复制到cuda 8.0文件夹。
装完以后运行,出现以下错误:
ImportError: No module named '_pywrap_tensorflow_internal' ImportError: DLL load failed: The specified module could not be found.
解决的办法是把cudnn64_6.dll改名为cudnn64_5.dll。
然后
import tensorflow
没有报错就成功了。