安装时出现导入错误:安装Horovod和Tensorflow时出错

Question

我正在尝试安装Tensorflow和Horovod

pip install tensorflow

HOROVOD_WITH_TENSORFLOW=1 pip install horovod

然后我运行了一个示例代码

import tensorflow as tf
import horovod.tensorflow as hvd

当我运行这段代码时，我得到以下错误

ImportError: Extension horovod.tensorflow has not been built.  If this is not expected, reinstall Horovod with HOROVOD_WITH_TENSORFLOW=1 to debug the build error.

Answer 1

如果您需要安装tensorflow和horovod，可以使用以下步骤：

1)创建conda环境，避免包版本不匹配。

conda create -n test_hvd -c intel python=3.6

2)开通环境

source activate test_hvd

(可以使用任何名称，而不是test_hvd，后者是环境名称。)

3)在开通环境下安装tensorflow：

pip install  https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-1.10.0-cp36-cp36m-linux_x86_64.whl

4)最后安装horovod

pip install --no-cache-dir horovod

注: 1.敬请确认您使用的是最新版本的GCC (gcc (GCC) 6.4.0运行正常)。

1. 在Linux OS

上测试了这些步骤

希望这能有所帮助！