Macにxformersをインストールする

環境

• macOS Sonoma 14.4.1 (Intel)
• pyenv local 3.10

問題

何も考えずにxformersをインストールしようとすると以下のようなエラーになる。

$ pip install xformers
clang: error: unsupported option '-fopenmp'

xformersの最新版ではCUDA以外の環境もサポートすべくOpenMPを導入したらしいが、MacのclangはOpenMPをサポートしていないためビルドエラーとなっているようです。

解決

# homebrewでgccをインストールする
$ brew install gcc
# venvを作る
$ python3 -m venv venv
# シンボリックリンクを貼る
$ ln -s /usr/local/bin/gcc-13 venv/bin/gcc
$ ln -s /usr/local/bin/g++-13 venv/bin/g++
$ ln -s /usr/local/bin/c++-13 venv/bin/c++
# venv起動
$ source venv/bin/activate
# xformersをインストール
$ pip install xformers
Successfully installed numpy-1.26.4 xformers-0.0.25.post1