[ubuntu] mpi4py 설치 시 에러

ubuntu에서 pip install mpi4py로 mpi4py를 설치할 때 다음과 같이 에러메세지가 나는 경우가 있다.

....
      error: Cannot compile MPI programs. Check your configuration!!!
      [end of output]

  note: This error originates from a subprocess, and is likely not a problem with pip.
  ERROR: Failed building wheel for mpi4py
Failed to build mpi4py
ERROR: Could not build wheels for mpi4py, which is required to install pyproject.toml-based projects

 

나의 경우는 pytorch/pytorch:2.1.1-cuda12.1-cudnn8-devel이 이미지를 사용한 컨테이너에서 나는 에러였다.

이때 which python으로 확인해보면 /opt/conda/bin/python을 사용하고 있다.

 

때문에 conda install -c conda-forge mpi4py mpich으로 설치해주면 잘 된다.

 

 

 

[출처]

• https://mpi4py.readthedocs.io/en/latest/install.html#using-conda