numthreads

Set the number of threads used by OpenBLAS, MKL, OMP, NumExpr, and Accelerate

numthreads is a concise, easy-to-use tool designed to set the number of threads for various computing libraries including OpenBLAS, Intel's Math Kernel Library (MKL), OpenMP, NumExpr, and Accelerate. This Python-based utility aids in optimizing the performance of numerical and scientific computing applications by allowing users to efficiently control the threading behavior of these key libraries.

• Simple and straightforward command-line interface.
• Sets thread count for OpenBLAS, MKL, OpenMP, NumExpr, and Accelerate.
• Context manager support for temporary thread setting in Python code.
• Cross-platform compatibility (Linux, macOS, Windows).
• No dependencies.

:books: Table of Contents

• :package: Installation
• :rocket: Quick Start
  • Unix-like Systems (Linux, macOS, WSL)
  • Windows (PowerShell)
  • Using as a Python Module
• :question: Getting Help

:package: Installation

To install numthreads, run the following command:

pip install "numthreads"

:rocket: Quick Start

After installing numthreads, you can easily set the number of threads used by supported libraries via the command line. For example, to set the number of threads to 4, run:

numthreads 4

Unix-like Systems (Linux, macOS, WSL)

To apply the settings in your shell:

eval $(numthreads <number_of_threads>)

Windows (PowerShell)

In PowerShell, use:

Invoke-Expression $(numthreads <number_of_threads>)

Using as a Python Module

You can also use numthreads as a Python module:

from numthreads import set_num_threads

set_num_threads(4)

or

from numthreads import num_threads

with num_threads(4):
    # Your code here will run with the specified number of threads
    pass

:question: Getting Help

For more information, or to report issues, please visit numthreads GitHub repository.