파이썬의 os 모듈은 운영체제와 상호작용하는 기능을 제공합니다. 이 중에서 os.cpu_count() 함수는 현재 시스템의 CPU 개수를 반환해주는 유용한 함수입니다.

사용법

os.cpu_count() 함수는 매우 간단하게 사용할 수 있습니다. 다음은 간단한 예제 코드입니다:

import os

cpu_count = os.cpu_count()
print(f"현재 시스템의 CPU 개수: {cpu_count}")

위 코드를 실행하면, 현재 시스템의 CPU 개수가 출력됩니다.

예제

다음은 os.cpu_count()를 사용하여 CPU 개수에 따라 병렬 처리를 수행하는 예제 코드입니다:

import os
import multiprocessing

def work(item):
    # 작업을 수행하는 함수
    print(f"작업 {item}을 처리 중...")
    # ...

if __name__ == "__main__":
    # CPU 개수 조회
    num_cpus = os.cpu_count()

    # 병렬 작업 수행을 위한 프로세스 풀 생성
    pool = multiprocessing.Pool(num_cpus)

    # 작업을 병렬로 처리
    items = [1, 2, 3, 4, 5]
    pool.map(work, items)

    # 프로세스 풀 종료
    pool.close()
    pool.join()

위 예제에서는 os.cpu_count() 함수를 사용하여 현재 시스템의 CPU 개수를 조회하고, 이를 이용하여 프로세스 풀을 생성합니다. 작업은 work() 함수로 정의되며, pool.map() 함수를 사용하여 병렬로 처리됩니다.

요약

os.cpu_count() 함수를 사용하면 파이썬 코드에서 현재 시스템의 CPU 개수를 조회할 수 있습니다. 이를 활용하여 병렬 처리 등 다양한 작업에 활용할 수 있습니다.