basecls.utils.env#

basecls.utils.env.set_nccl_env()[源代码]#

Set NCCL environments, which is essential to multi-node training.

basecls.utils.env.set_num_threads(num=1)[源代码]#

Set number of threads in OpenMP, OpenCV, MKL, OPENBLAS, VECLIB, NUMEXPR, etc.

参数

num (int) – number of threads. Default: 1