My computer has 8 CPU-kernels, when I use CPU to run resnet50 using TVM, I found only one cpu is running, is it possible to use all the 8 cpu-kernels at the same time?
What machine is it? You may need to specify TVM_NUM_THREADS=8
at runtime.
2 Likes
Note that in the case that your machine has 8 logical cores but only 4 physical cores, we typically set the number of threads to 4.
1 Like
I use following code to configure TVM_NUM_THREADS:
num_threads = 4
os.environ["TVM_NUM_THREADS"] = str(num_threads)
the result performance doesn’t change at all.
can you help me ?
1 Like
I am not sure how often TVM resets the number of threads used in the runtime if it is set dynamically in Python. To be sure, can you export TVM_NUM_THREADS=4
before running your script?
1 Like
I also see this problem. I have tries to export TVM_NUM_THREADS before running my script and it does not work…
1 Like