为了账号安全,请及时绑定邮箱和手机立即绑定

tensorflow 怎样确定是cpu 还是 gpu?

tensorflow 怎样确定是cpu 还是 gpu?

米琪卡哇伊 2018-10-17 19:14:15
tensorflow 怎样确定是cpu 还是 gpu
查看完整描述

1 回答

?
尚方宝剑之说

TA贡献1788条经验 获得超4个赞

安装GPU版本的tensorflow就可以了pip install --upgrade 还需要安装GPU加速包,下载 CUDA 和 cuDNN 的安装包了,注意版本号分别是 CUDA 8.0 和 cuDNN 5.1。
如果都安装成功,运行程序时会自动使用GPU计算

查看完整回答
反对 回复 2018-11-11
  • 1 回答
  • 0 关注
  • 2445 浏览

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信