tensorflow 如何控制使用cpu数量

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:

猜你喜欢

转载自blog.csdn.net/yeyang911/article/details/84647015