时间戳

首页 > 解决方案 > 安装导入错误：安装 Horovod 和 Tensorflow 时出错

tensorflow - 安装导入错误：安装 Horovod 和 Tensorflow 时出错

问题描述

我正在尝试安装 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.

标签： tensorflow

解决方案

如果需要安装 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 操作系统上进行了测试

希望这可以帮助！

推荐阅读