tensorflow读取数据有时候非常占用资源,针对这一情况,我们可以通过控制程序使用CPU的数量来限制
步骤如下:
1、bashrc 中修改指定个数
$ export CPU_NUM=2
2、python代码中在需要运行的程序前加入如下配置代码
cpu_num = int(os.environ.get('CPU_NUM',1))
config = tf.ConfigProto(device_count={"CPU": cpu_num},
inter_op_parallelism_threads = cpu_num,
intra_op_parallelism_threads = cpu_num,
log_device_placement=True)
with tf.Session(config = config) as sess: