From e4c5d3526f6eefe352892ef1293d062f7aecb09b Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Fri, 24 Jul 2020 21:48:58 +0200 Subject: [PATCH] Issue 9631: Tf1.14 does not have tf.config.list_physical_devices. (#9681) --- rllib/evaluation/rollout_worker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rllib/evaluation/rollout_worker.py b/rllib/evaluation/rollout_worker.py index e2dd3234e..c3a12ba52 100644 --- a/rllib/evaluation/rollout_worker.py +++ b/rllib/evaluation/rollout_worker.py @@ -412,7 +412,7 @@ class RolloutWorker(ParallelIteratorWorker): worker_index) + " on CPU (please ignore any CUDA init errors)") elif (policy_config["framework"] in ["tf2", "tf", "tfe"] and - not tf.config.list_physical_devices("GPU")) or \ + not tf.config.experimental.list_physical_devices("GPU")) or \ (policy_config["framework"] == "torch" and not torch.cuda.is_available()): raise RuntimeError(