Solved by ptrblck in post #2 
              
              
                You could use torch.set_num_threads(int) to define the number of threads used for intraop parallelism and torch.set_num_interop_threads(int) for interop parallelism (e.g. in the JIT interpreter) on the CPU. Also, the env vars OMP_NUM_THREADS and MKL_NUM_THREADS might be useful. 
More information are…