是否可以通过4个CPU上的多处理来拟合Keras模型? (没有GPU) 如果是这样,该如何处理?
digohao 回答:是否可以通过CPU多处理来拟合Keras模型?
默认情况下,Keras将尝试使用计算机上可用的所有内核并行(多处理)拟合模型。
您可以使用以下代码段选择cpus(或作业)的数量:
from keras import backend as K
import tensorflow as tf
jobs = 4 # number of cores
config = tf.ConfigProto(intra_op_parallelism_threads=jobs,inter_op_parallelism_threads=jobs,allow_soft_placement=True,device_count={'CPU': jobs})
session = tf.Session(config=config)
K.set_session(session)