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。
这里写图片描述
差点忘记了,还有一点很重要就是cudnn的版本也很重要。一定要看好tensorflow给出的错误提示中dll前面的版本号,版本不对拷过去也没用。我的错误提示显示缺少cudnn64_7.dll,结果我第一次尝试的时候装的是cudnn64_70.dll,(也是从csdn下载的,似乎版本不对,但文件名真的很像啊。。)就这样都不行。所以一定要严格按照版本来!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值