加载中…
个人资料
  • 博客等级:
  • 博客积分:
  • 博客访问:
  • 关注人气:
  • 获赠金笔:0支
  • 赠出金笔:0支
  • 荣誉徽章:
正文 字体大小:

如何使Keras调用GPU?

(2017-04-20 10:59:13)
标签:

学习

分类: Keras

资料来源于 http://keras-cn.readthedocs.io/en/latest/for_beginners/FAQ/


如果采用TensorFlow作为后端,当机器上有可用的GPU时,代码会自动调用GPU进行并行计算。如果使用Theano作为后端,可以通过以下方法设置:

方法1:使用Theano标记

在执行python脚本时使用下面的命令:


THEANO_FLAGS=device=gpu,floatX=float32 python my_keras_script.py

方法2:设置.theano文件

点击这里查看指导教程

方法3:在代码的开头处手动设置theano.config.devicetheano.config.floatX


    import theano
    theano.config.device = 'gpu'
    theano.config.floatX = 'float32'


0

阅读 收藏 喜欢 打印举报/Report
  

新浪BLOG意见反馈留言板 欢迎批评指正

新浪简介 | About Sina | 广告服务 | 联系我们 | 招聘信息 | 网站律师 | SINA English | 产品答疑

新浪公司 版权所有