[rllib] Restore TF soft placement config to fix multi-GPU optimizer (#2395)

2026-06-28 22:20:52 +08:00 · 2018-07-13 10:34:37 +02:00
parent d6af50785e
commit 62f84d2f07
1 changed files with 1 additions and 0 deletions
@@ -46,6 +46,7 @@ COMMON_CONFIG = {
        "gpu_options": {
            "allow_growth": True,
        },
+        "allow_soft_placement": True,  # required by PPO multi-gpu
    },
    # Whether to LZ4 compress observations
    "compress_observations": False,