系统跑着跑着就崩溃了,不动了,可以如此设置一下:
使用这些参数,可以达到最大150%的加速
from keras import backend as K
import tensorflow as tf
NUM_PARALLEL_EXEC_UNITS = 6
config = tf.ConfigProto(intra_op_parallelism_threads = NUM_PARALLEL_EXEC_UNITS,
inter_op_parallelism_threads = 1,
allow_soft_placement = True,
device_count = {'CPU': NUM_PARALLEL_EXEC_UNITS })
session = tf.Session(config=config)
K.set_session(session)
import os
os.environ["OMP_NUM_THREADS"] = str(NUM_PARALLEL_EXEC_UNITS)
os.environ["KMP_BLOCKTIME"] = "30"
os.environ["KMP_SETTINGS"] = "1"
os.environ["KMP_AFFINITY"]= "granularity=fine,verbose,compact,1,0"