import tensorflow as tf import multiprocessing as mp core_num = mp.cpu_count() config = tf.ConfigProto( inter_op_parallelism_threads=core_num, intra_op_parallelism_threads=core_num ) a = tf.random_uniform([1], minval=1, maxval=10) ss= tf.Session(config=config) print(ss.run(a))