To Top
首页 > 常用平台 > 正文

keras使用多核cpu

标签:keras, theano, 多核cpu, htop


目录

使用theano后端

开启theano后端

vim ~/.keras/keras.json,修改为

{
    "epsilon": 1e-07,
    "floatx": "float32",
    "image_data_format": "channels_last",
    "backend": "theano"
}

打开openmp的flag

import theano
theano.config.openmp = True

确保gcc版本

至少要是482的

export PATH=/opt/compiler/gcc-4.8.2/bin/:$PATH

设置环境变量并运行

OMP_NUM_THREADS=20 python xxx.py

使用tensorflow后端

参考:Tensorflow并行:多核(multicore),多线程(multi-thread)

用htop查看进程情况

jumbo install htop

htop样例如下:


原创文章,转载请注明出处!
本文链接:http://daiwk.github.io/posts/platform-keras-multi-cpu.html
上篇: perl
下篇: 各框架的embedding实现

comment here..