[rllib] Raise an error if GPUs are enabled but not tf.test.is_gpu_available() (#6365)

This commit is contained in:
Eric Liang
2019-12-05 10:13:54 -08:00
committed by GitHub
parent 668ce47360
commit 4c6739476b
3 changed files with 13 additions and 15 deletions
+1 -1
View File
@@ -118,7 +118,7 @@ class RayTrialExecutor(TrialExecutor):
# Clear the Trial's location (to be updated later on result)
# since we don't know where the remote runner is placed.
trial.set_location(Location())
logger.info("Trial %s: Setting up new remote runner.", trial)
logger.debug("Trial %s: Setting up new remote runner.", trial)
# Logging for trials is handled centrally by TrialRunner, so
# configure the remote runner to use a noop-logger.
return cls.remote(config=trial.config, logger_creator=logger_creator)