Only one core is used in cpu mode.
psychosomaticdragon opened this issue · 5 comments
Current environment: windows 7, conda using python 3.6, fx-8350 cpu.
When i test using torch.get_num_threads() it returns 1. I tried fixing with
torch.set_num_threads(8)
and the number returned by get_num_threads() updates - however pytorch still only uses one thread. I think it has to be specific to the windows port, because running pytorch within a virtual linux environment utilizes all available cores in the virtual machine.
@psychosomaticdragon This build is compiled with MKL. However, you are using AMD CPU. I think that may be this issue. Would you mind building with source and see if the issue persists?
i'll give it a shot, i'm not too familiar with windows builds however - can you point me in the right direction for info on how to build it? Thanks.
See Compiling Section of README.md.
#6416 in pytorch/pytorch is on this now.
It is landed in master.