How to make gcn run only using 1 thread and 4 thread?
AbhishekUoR opened this issue · 0 comments
AbhishekUoR commented
Currently, GCN on CPU is using all threads of the machine, I use the following command "ps -o nlwp " to know how many threads it is using, the number comes out to be 51 on 48 hardware threads machine
I want to make it run only on 1 thread and 4 thread to see its performance, I tried inserting following TensorFlow code in train.py file
"""
session_conf = tf.ConfigProto(intra_op_parallelism_threads=1, inter_op_parallelism_threads=1)
sess = tf.Session(config=session_conf)
"""
But still, it uses 51 threads, is there any part of code in gcn directory wherein I can control the number of threads?