how do I disable multiprocessing?

[alita-moore]

as far as I can tell albumentations automatically performs multiprocessing, but I want to run one augmentation per cpu core (I want to manage the multiprocessing myself). Is it possible to disable multiprocessing for albumentations?

[ternaus]

Albumentations by itself does not perform multiprocessing, but libraries it depends on could, like opencv or numpy.

import os
os.environ["OMP_NUM_THREADS"] = "1"
os.environ["OPENBLAS_NUM_THREADS"] = "1"
os.environ["MKL_NUM_THREADS"] = "1"
os.environ["VECLIB_MAXIMUM_THREADS"] = "1"
os.environ["NUMEXPR_NUM_THREADS"] = "1"

should do the job.

[alita-moore]

setting these variables globally when calling the script i.e. OMP_NUM_THREADS=1 python example.py worked