我有一个处理1000张图像的管道。每个图像都要经过4个步骤，以创建模型的输入。因此，有4000个数据准备任务以及一个最终预测任务。总共4001个任务。
luigi并行处理了4000个数据准备任务，因此4个cpus上可以同时运行4个任务。为此，我设置OMP_THREAD_LIMIT = 1，否则由于luigi和OMP之间的冲突而挂起。
最终预测任务使用pytorch。这是一个单一的luigi任务，但通过pytorch通过OMP进行并行处理。因此，我在开始任务之前重置OMP_THREAD_LIMIT。
这可行，但是在前4000个任务中，我收到成千上万的警告消息.....“ OMP：警告＃96无法使用1代替4个线程组成一个团队”和“ OMP：提示考虑取消设置。 .... OMP_THREAD_LIMIT”。
如何禁用这些消息？还是有其他方法可以在没有OMP_THREAD_LIMIT的情况下暂时禁用OMP？