Merge pull request tensorflow#63094 from tensorflow/r2.16-20a8fc4339d

learning-to-play · web-flow · commit 7c75122d0669 · 2024-02-29T11:41:22.000-08:00
r2.16 cherry-pick: 20a8fc4 "Inherit env variables for tpu grpc worker console script"
diff --git a/tensorflow/python/tools/grpc_tpu_worker_service.py b/tensorflow/python/tools/grpc_tpu_worker_service.py
@@ -74,6 +74,7 @@ def move_file_to_systemd(service_name):
 
 def enable_start_service(service_name):
   commands = [
+      "systemctl --user import-environment",
       "systemctl --user daemon-reload",
       f"systemctl --user enable {service_name}",
       f"systemctl --user start {service_name}",
