Hello everyone.
I recently updated to python3 and I am seeing that I cannot set the number of omp proccess anymore in pytorch.
This is the check I did:
python2.7
import torch
print(torch.__version__)
>1.0.0.dev20181003
print(torch.get_num_threads())
> 20
torch.set_num_threads(8)
print(torch.get_num_threads())
> 8
Here everything works as expected. When I run exactly the same with python3.7 I get:
import torch
print(torch.__version__)
>1.0.0
print(torch.get_num_threads())
> 1
torch.set_num_threads(8)
print(torch.get_num_threads())
> 1
Any clue why is this happening?
Using envirionment variable OMP_NUM_THREADS
doesn’t work any better.
Thank you very much in advance.