如何使Keras调用GPU?
(2017-04-20 10:59:13)
标签:
学习 |
分类: Keras |
资料来源于
如果采用TensorFlow作为后端,当机器上有可用的GPU时,代码会自动调用GPU进行并行计算。如果使用Theano作为后端,可以通过以下方法设置:
方法1:使用Theano标记
在执行python脚本时使用下面的命令:
THEANO_FLAGS=device=gpu,floatX=float32 python my_keras_script.py
方法2:设置.theano
文件
点击这里查看指导教程
方法3:在代码的开头处手动设置theano.config.device
和theano.config.floatX
import theano
theano.config.device = 'gpu'
theano.config.floatX = 'float32'
前一篇:batch epoch
后一篇:如何保存Keras模型?