https://jdhao.github.io/2020/07/06/pytorch_set_num_threads/