Ogranicz liczbę wątków w numpy

Question

Wygląda na to, że moja biblioteka numpy używa 4 wątków, a ustawienie OMP_NUM_THREADS=1 nie zatrzymuje tego.

numpy.show_config() daje mi te wyniki:

atlas_threads_info: 
    libraries = ['lapack', 'ptf77blas', 'ptcblas', 'atlas'] 
    library_dirs = ['/usr/lib64/atlas'] 
    define_macros = [('ATLAS_INFO', '"\\"3.8.4\\""')] 
    language = f77 
    include_dirs = ['/usr/include'] 
blas_opt_info: 
    libraries = ['ptf77blas', 'ptcblas', 'atlas'] 
    library_dirs = ['/usr/lib64/atlas'] 
    define_macros = [('ATLAS_INFO', '"\\"3.8.4\\""')] 
    language = c 
    include_dirs = ['/usr/include'] 
atlas_blas_threads_info: 
    libraries = ['ptf77blas', 'ptcblas', 'atlas'] 
    library_dirs = ['/usr/lib64/atlas'] 
    define_macros = [('ATLAS_INFO', '"\\"3.8.4\\""')] 
    language = c 
    include_dirs = ['/usr/include'] 
openblas_info: 
    NOT AVAILABLE 
lapack_opt_info: 
    libraries = ['lapack', 'ptf77blas', 'ptcblas', 'atlas'] 
    library_dirs = ['/usr/lib64/atlas'] 
    define_macros = [('ATLAS_INFO', '"\\"3.8.4\\""')] 
    language = f77 
    include_dirs = ['/usr/include'] 

Tak wiem, że to jest za pomocą Blas, ale nie mogę dowiedzieć się, jak zrobić to użyć 1 Temat dla mnożenia macierzy.

Answer 1

spróbuj ustawić wszystkie z poniższych kryteriów:

export MKL_NUM_THREADS=1 export NUMEXPR_NUM_THREADS=1 export OMP_NUM_THREADS=1

Czasami jest to trochę skomplikowane, aby zobaczyć, gdzie dokładnie jest wprowadzany wielowątkowość.