haoyuhu/bert-multi-gpu

ImportError: No module named 'tensorflow.python.distribute.cross_device_ops'

LeoWood opened this issue · 6 comments

LeoWood commented

When I ran the script run_custom_classifier.py, the error occured:
ImportError: No module named 'tensorflow.python.distribute.cross_device_ops'

python: 3.5.2
tensorflow-gpu: 1.11.0
ubuntu 16.04

How can I handle this?
Thanks

haoyuhu commented

Please try upgrading TensorFlow to 1.14.

  • 👍1
stale commented

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

shishishu commented

Please try upgrading TensorFlow to 1.14.

Yep, it succeed with this suggestions. Thanks Haha.

  • 🎉1
bigzhouj commented

Tensorflow version 1.15does not work，only 1.14?

haoyuhu commented

Tensorflow version 1.15does not work，only 1.14?

TF verison above 1.14 is ok.

  • 👍1