keras使用GPU加速计算

判断

本人机子内存8G,显卡GTX1060 6G显存,用 keras 跑小例子时候也能跑,也没发现自己运行时到底是用的CPU还是GPU,这次跑了稍微长点的例子,然后 jupyter notebook 内核就死了,就想起来验证一下到底用的是什么,网上查了下验证方式,感觉不怎么好,还是直接看运行时机子的情况,打开电脑的控制台或者任务管理器,看运行时GPU显存占的百分比:
这里写图片描述
像图片上这样 0% 的就是没用到GPU计算~

解决

keras 使用GPU是基于 tensorflow 的GPU版本,所以使用 keras 要先安装tensorflow-gpu
使用 pip list ; conda list 两条指令可以分别看到用 pipconda 安装的包,可以使用
pip uninstall 包name
conda uninstall 包name 分别卸载之前安装错的包

安装

这里只建议使用 pip 方法安装 tensorflow-gpu 和 keras ,因为有看到博客上说,使用 conda 安装时会默认安装CPU版本(就是因为这句话,所以我就把 conda 安装的全卸了,用 pip 重新安装了下,然后就可用了)
pip install tensorflow-gpu
pip install keras
可完成安装~
可用上述 判断 里的在运行时看到使用 GPU 加速计算~
这里写图片描述

  • 2
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 8
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值