控制tensorflow使用的核数

BiddingKG/dl/interface/predictor.py

@@ -27,7 +27,7 @@ import calendar
 import datetime
 # import fool   # 统一用 selffool ，阿里云上只有selffool 包
 
-cpu_num = int(os.environ.get("CPU_NUM"))
+cpu_num = int(os.environ.get("CPU_NUM",0))
 sess_config = tf.ConfigProto(
                         inter_op_parallelism_threads = cpu_num,
                         intra_op_parallelism_threads = cpu_num,