keras deep learning use gpu

来源:互联网 发布:信息检索系统知乎 编辑:程序博客网 时间:2024/06/06 07:45

keras is a convinient deep learing package. so i try to use it to apply lstm. but if you want keras use gpu ,you need to config ~/.theanorc

[global]floatX = float32device = gpu0[lib]cnmem = 1

 and the time before training is started is a little bit longer than using cpu.

0 0
原创粉丝点击