Search code examples
dockertensorboard

Run tensorboard to visualise docker container running reinforcement learning in Ray Rllib

I have a docker container running a custom simulation for reinforcement learning. I have placed all of this code in a docker container.

This code takes a long time to run, and I would like to be able to visualise the algorithm in real time.

However, I am unsure on how to start tensorboard from this container. I was thinking of saving the data to a volume and then accessing that from another container.

Solution

  • I ended up doing as a I proposed, saving the data in a volume and accessing the external data source via tensorboard.

    docker run -it -v <path/to/tensorflow_data>:</path/to/vol> -v <docker_container>:latest

    Where code inside <> should be replaced with where the tensorflow data is stored, the path where you saved the volume inside the container and the name of your container.