ubuntu 18.04下使用keras+tensorflow-gpu(安装cuda,cudnn)

在经历了接近两天的瞎搞中,终于完成了ubuntu下用gpu训练模型,真是比cpu快了好多倍啊。。。

先看看成功结果图

可以看到这时的gpu平均利用率达到了80%以上,而cpu利用率保持在20%以下。
成功的标志:在python/python3环境中导入tensorflow-gpu成功(可以用 tensorflow._ _version __查看版本)。或者查看系统CPU和GPU使用情况。

必须强调的是

要想成功,版本对的上是关键,你若对不上,那就不可能成功
需要明白的是:tensorflow-gpu应该是会调用cuda提供的函数(或者说tensorflow-gpu通过cuda才能在gpu上运行),所以tensorflow的版本和cuda的版本需要对应;而cudnn又是建立在cuda上的,所以cudnn需要和cuda版本对应;同时keras往往使用tensorflow作为backend,所以keras需要和tensorflow-gpu版本对应(我估计若同时装有tensorflow和tensorflow-gpu的情况下,keras会优先利用tensorflow而不是tensorflow-gpu,所以请只保留tensorflow-gpu,反正多一

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值