/docker_mldm_gpu

my machine learning docker-compose setup, with gpu support on ec2 g-instainces.

Primary LanguageRoff

Docker MLDM-GPU

my docker environment for machine learning with GPU support.

How to use

just follow the script number,
turn an empty instance to a good-to-go deep-learning environment in 30 mins.

  • 0-prepare-host.sh
    install docker, nvidia driver, nvidia-docker in your host machine.

  • 1-build-mldm.sh
    build my machine learning toolbelt container
    here I choose tensorflow gpu supported version for deep learning, you may append whatever you like in Dockerfile!

  • 2-start-mldm.sh
    start the container with some environment setup script, with ipython notebook in http://<your_machine>:8888

Enjoy!