1. 程式人生 > >tensorflow 如何控制使用cpu數量

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: