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可以正常工作）。

这些步骤已在Linux OS上进行了测试

希望这会有所帮助！

安装中的导入错误：安装Horovod和Tensorflow时出错

1 个答案: