Dockerfiles for building docker images with a pre-prepared environment for agent trainer.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
Dockerfile.cpu
Dockerfile.gpu
LICENSE
README.md
supervisord.conf

README.md

Dockerfiles for Agent Trainer

Dockerfiles for building docker images with a pre-prepared environment for agent trainer. More details in this blogpost.

Usage

Two flavors of images are provided, one based on Tensorflow's CPU image and the other its GPU flavor

# CPU flavor
docker build -f Dockerfile.cpu -t lopespm/agent-trainer-cpu /folder/of/cloned/dockerfiles
# GPU flavor
docker build -f Dockerfile.gpu -t lopespm/agent-trainer-gpu /folder/of/cloned/dockerfiles