限制OpenMP线程数提速
背景:视频帧序列逐个跑一遍CNN模型,发现CPU被占满了经过debug发现问题出在torchvision transorms.totensor上经过搜索,torch在设计多线程加速计算时候会默认占满所有core,因此cpu会飙满,这对别的应用程序速度影响非常大,因此需要控制一下。解决方法:1)torch.set_num_threads(1)手动控制一下torch占用的线程数2)设置环境变量export OMP_NUM_THREADS=1 or export MKL_NUM_THREADS=1