https://github.com/pytorch/pytorch/issues/33296
OMP_NUM_THREADS=1 MKL_NUM_THREADS=1 python train.py
or
import os
os.environ["OMP_NUM_THREADS"] = "1"
os.environ["MKL_NUM_THREADS"] = "1"
https://github.com/pytorch/pytorch/issues/33296
OMP_NUM_THREADS=1 MKL_NUM_THREADS=1 python train.py
or
import os
os.environ["OMP_NUM_THREADS"] = "1"
os.environ["MKL_NUM_THREADS"] = "1"