python - 在哪里使用光线 Actor 导入 tensorflow ？

Question

使用光线 Actor 并行运行多个 tensorflow 模型，我问自己在哪里导入 tensorflow :

# [1] maybe import tensorflow here?

@ray.remote(num_cpus=1)
class Remote_Runner:
    # [2] maybe import tensorflow here?
    def __init__(self, weights):
        # [3] maybe import tensorflow here?
        self.model=My_model()
        self.model.set_weights(wegihts)

    def do_something_with_model:
        self.model.do_something()

文档中给出的示例提到了由于“导入 TensorFlow 和设置全局状态的副作用”而在 Actor 中导入 Tensorflow，但仅给出了射线远程函数的示例。那么我应该在 [1]、[2] 或 [3] 甚至其他地方运行“import tensorflow as tf”？这里是否有可以遵循的最佳实践，[1]、[2]和[3]之间有什么区别，即在每种情况下如何访问 tensorflow 以及它们何时执行？

Answer 1

这个问题现在可能已经修复，但最安全的选择是在 [3] 处导入 tensorflow 。