Mac OS 에서 PyTorch 사용할 때 NumPy 에러 해결

Huggping Face를 사용하기 위해서 PyTorch가 필요해서 MacOS에서 설치를 해보려했다. 현재 사용중인 우선 virtualenv 를 사용하여 테스트할 공간을 만들어서 진행한다.

virtualenv huggingface

테스트할 디렉토리가 만들어지면 virtualenv activate로 가상환경을 활성화 한다.

source ./bin/activate

PyTorch를 설치한다.

pip install torch torchvision torchaudio

가상환경이 활성화되면 Python 인터프리터로 들어가기 위해서 python 을 실행한다.

python

앞에서 설치한 torch가 정상적으로 잘 설치되고 사용 가능한지 알아보기 위해서 다음 코드를 입력한다.

import torch

정상적인 실행 결과는 torch를 임포트하고 다음 프롬프트가 나와야 하지만 NumPy 관련 에러가 발생했다.

A module that was compiled using NumPy 1.x cannot be run in
NumPy 2.1.2 as it may crash. To support both 1.x and 2.x
versions of NumPy, modules must be compiled with NumPy 2.0.
Some module may need to rebuild instead e.g. with 'pybind11>=2.12'.

If you are a user of the module, the easiest solution will be to
downgrade to 'numpy<2' or try to upgrade the affected module.
We expect that some modules will need time to support NumPy 2.

이 에러는 NumPy 2.x 버전 보다 낮은 버전이 사용하는데 현재 설치되어 있는 NumPy가 2.1.2가 설치되어 있어 발생하는 에러이다.

NumPy를 2.0 이하로 다시 설치한다.

pip install "numpy<2.0"

NumPy를 호환성에 맞게 2.0 보다 낮은 버전을 설치한 후 다시 python 인터프리터로 들어가서 확인하면 torch를 정상적으로 사용할 수 있는 것을 확인할 수 있다.