horovod.tensorflow

[Jobiro-Jostar]

When I typed import horovod.tensorflow, it occured an error like that:
Traceback (most recent call last):
File "", line 1, in
File "/adddisk/zhangjin/projects/horovod/horovod/tensorflow/init.py", line 25, in
check_extension('horovod.tensorflow', 'HOROVOD_WITH_TENSORFLOW', file, 'mpi_lib')
File "/adddisk/zhangjin/projects/horovod/horovod/common/util.py", line 51, in check_extension
'Horovod with %s=1 to debug the build error.' % (ext_name, ext_env_var))
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.

[chenguoguo]

@Some-random could you take a look or assign it to someone else? I didn't encounter this in my environment...

[Some-random]

@Some-random could you take a look or assign it to someone else? I didn't encounter this in my environment...

I'm looking into this. @8z8z8z wrote most code for horovod part, can you also take a look at this?

[JianweiSun007]

@Jobiro-Jostar Hi, your environment maybe have some issues.
Maybe this can help you
horovod/horovod#314
and you can try this command:

TensorFlow
To ensure that Horovod is built with TensorFlow support enabled:

$ HOROVOD_WITH_TENSORFLOW=1 pip install horovod[tensorflow]
or
HOROVOD_WITH_TENSORFLOW=1 [other flags you used before] pip install -v --no-cache-dir horovod

and more detail you can get form,
https://github.com/horovod/horovod/blob/master/docs/install.rst

[Jobiro-Jostar]

@Jobiro-Jostar Hi, your environment maybe have some issues.

Maybe this can help you
horovod/horovod#314
and you can try this command:

TensorFlow
To ensure that Horovod is built with TensorFlow support enabled:

$ HOROVOD_WITH_TENSORFLOW=1 pip install horovod[tensorflow]

or
HOROVOD_WITH_TENSORFLOW=1 [other flags you used before] pip install -v --no-cache-dir horovod

and more detail you can get form,
https://github.com/horovod/horovod/blob/master/docs/install.rst
I typed the command:HOROVOD_WITH_TENSORFLOW=1 [other flags you used before] pip install -v --no-cache-dir horovod[tensorflow] and the error was fixed, many thx to you