streamlit_kmeans 진행중 경고 - UserWarning

https://github.com/reodinas/streamlit_kmeans

프로젝트 진행 중 발생한 문제 해결 과정을 기록한 포스트입니다.

 

 

sklearn.cluster.kmeans를 사용하다 뜬 warning.

사용 가능한 스레드보다 청크가 적을 때 MKL이 있는 Windows에서 메모리 누수가 발생하는 것으로 알려져 있고, 환경변수 OMP_NUM_THREADS=1 을 설정하여 이 문제를 방지할 수 있다고 한다.

 

OMP_NUM_THREADS=1 설정하는 방법

KMeans를 임포트하기 전에 아래 코드를 실행하면 더 이상 경고문이 뜨지 않는다.

 

import os
os.environ["OMP_NUM_THREADS"] = "1"