macos - 在 MacOS 上安装 horovod

Question

通过安装 horovod 后pip3 install horovod

我收到一个错误：

ImportError: Extension horovod.tensorflow has not been built: /usr/local/lib/python3.7/site-packages/horovod/tensorflow/mpi_lib.cpython-37m-darwin.so not found
If this is not expected, reinstall Horovod with HOROVOD_WITH_TENSORFLOW=1 to debug the build error.

也试过

pip3 uninstall horovod
HOROVOD_WITH_TENSORFLOW=1 pip3 install horovod

pip3 uninstall horovod
HOROVOD_WITHOUT_MPI=1 pip3 install horovod

但得到同样的错误。

Answer 1

关于 Pip 的一个棘手的事情是，您需要指定--no-cache-dir以防止它重用其“缓存”版本的包（在您的情况下，使用不同的安装标志编译 Horovod。所以尝试这样的事情：

HOROVOD_WITHOUT_MPI=1 HOROVOD_WITH_TENSORFLOW=1 pip3 install --no-cache-dir horovod

特定于 macOS，您需要确保libuv在使用 Gloo（不带 MPI）安装 Horovod 之前已安装：

brew install libuv