Databricksでgensimを利用したい

現在起こっていること

Databricksでitem2vecを利用したいと思っています。
以下のようにインストールし

%pip install --upgrade pip %pip install --upgrade setuptools %pip uninstall numpy %pip install numpy dbutils.library.restartPython() %pip install gensim==4.0.1 dbutils.library.restartPython()

とすると

alledProcessError: Command 'pip --disable-pip-version-check install gensim==4.0.1' returned non-zero exit status 1.

となり、

import gensim

インポートすると

ValueError: numpy.ndarray size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject

となります。

ちなみにgensimのバージョンを指定しない場合、インストール時のエラーは出なくなりますが、import時のエラーはでます。