From 62f84d2f07a3020b3baf636d72cef7ff1179f8d9 Mon Sep 17 00:00:00 2001
From: Eric Liang <ekhliang@gmail.com>
Date: Fri, 13 Jul 2018 10:34:37 +0200
Subject: [PATCH] [rllib] Restore TF soft placement config to fix multi-GPU
 optimizer (#2395)

---
 python/ray/rllib/agents/agent.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/ray/rllib/agents/agent.py b/python/ray/rllib/agents/agent.py
index 454654b2c..cdd6bdee7 100644
--- a/python/ray/rllib/agents/agent.py
+++ b/python/ray/rllib/agents/agent.py
@@ -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,