win10 x64 安装tensorflow-gpu

因为最近在学习keras,发现用cpu比较慢,关键是发热太大了,所以想试试用gpu。我说用tensorflow作为后端的,之前tensorflow-gpu不支持win10,所以装的是cpu版本的,现在想更新成gpu版本。这里为了简单把安装过程记录如下。我的环境是win10 64,python3.6
首先直接

pip install tensorflow-gpu

我这一步没什么问题,但是进入python然后import tensorflow 的时候回报错,回提示缺少cudart64_90.dll之类的错误,这是因为没有安装cuda或者安装的cuda的版本不对。注意这里版本很重要很重要!主要是看错误提示后面的数字。比如我安装的时候提示cudart64_90.dll,这说明需要安装的是cuda9.0版本才行。我第一次就是因为直接下载了最新的9,1版本所以才安装失败的。记住版本一定要对上。
安装完cuda以后,在运行,还会报错,也是缺少个什么什么dll,一般是因为没有安装cudnn。可以直接去官网下载。https://developer.nvidia.com/rdp/cudnn-download这里是下载地址。 我比较惨,点进去结果官网正在维护,没办法只好在网上搜,结果还真找到了http://download.csdn.net/download/u012552296/10265295 我试了下,安装成功。资源没问题。
下载完cudnn解压缩,然后把里面的三个文件夹里面的内容分别添加至之前安装的cuda的相应位置。cuda的路径一般是在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin 弄完了以后还要主要要把文件夹中的bin以及lib/x64都添加到path路径中。这样配置好了以后就没有问题了。
用keras跑了个minst-cnn的example试了下,没问题,用gpu速度比我之前用cpu跑大概快了一半的样子,没用预想的多。大概是因为我的显卡不太好吧。我用的是笔记本,显卡是gtx965m。
win10 x64 安装tensorflow-gpu_第1张图片
差点忘记了,还有一点很重要就是cudnn的版本也很重要。一定要看好tensorflow给出的错误提示中dll前面的版本号,版本不对拷过去也没用。我的错误提示显示缺少cudnn64_7.dll,结果我第一次尝试的时候装的是cudnn64_70.dll,(也是从csdn下载的,似乎版本不对,但文件名真的很像啊。。)就这样都不行。所以一定要严格按照版本来!

你可能感兴趣的:(tensorflow,keras,备忘)