分布式训练cannot assign a device for operation PyFunc

[zhixiongning]

在跑learning/examples/tf/graphsage/dist_train.py 时training过程报cannot assign device for operation PyFunc: {{node PyFunc} was explicitly assigned to /job:localhost/replica:0/task:0 but available devices are [/job:worker/replica:0:task:0/device:CPU:0]. Make sure the device specification refers to a valid device.
我是在一台机器上启多个端口模拟分布式计算，未使用GPU

[sighingnow]

Please try to seek to tensorflow from similar issue reports (e.g., tensorflow/tensorflow#31318 ) and see if it is helpful for your case.

[zhixiongning]

我参照相关issue尝试了一些办法还是没有解决。
另外，我参考这个https://github.com/tensorflow/tensorflow/issues/52959，修改了tensorflow的版本2.2.0(我之前用的最新版2.8.0)，但又遇到了grpc的错误。想问下当时这个分布式example是基于tensorflow的哪个版本运行的？谢谢

[sighingnow]

我参照相关issue尝试了一些办法还是没有解决。 另外，我参考这个https://github.com/tensorflow/tensorflow/issues/52959，修改了tensorflow的版本2.2.0(我之前用的最新版2.8.0)，但又遇到了grpc的错误。想问下当时这个分布式example是基于tensorflow的哪个版本运行的？谢谢

我们的CI中测试过的是tensorflow 2.5.2

[zhixiongning]

我参照相关issue尝试了一些办法还是没有解决。 另外，我参考这个https://github.com/tensorflow/tensorflow/issues/52959，修改了tensorflow的版本2.2.0(我之前用的最新版2.8.0)，但又遇到了grpc的错误。想问下当时这个分布式example是基于tensorflow的哪个版本运行的？谢谢

我们的CI中测试过的是tensorflow 2.5.2

谢谢，我使用2.2.0解决了这个问题
在import tensorflow之前我加了os.environ.pop('http_proxy')，os.environ.pop('https_proxy')

[sighingnow]

Close as resolved.