Training a neural network with Docker
This the code I used in my article A Practical example of Training a Neural Network in the AWS cloud with Docker.
I prepared also ready-to-use Docker Images on Docker Hub.
Setup of a AWS instance
To train the neural network with GPU power on AWS you need to set up instance with docker-machine. The ami-0891f5dcc59fc5285 is a AMI I am sharing to the public. It has CUDO 10.1 and nvidia docker, which is needed to "activate" the GPU.
docker-machine create --driver amazonec2\ --amazonec2-instance-type p2.xlarge\ --amazonec2-ami ami-0891f5dcc59fc5285\ --amazonec2-vpc-id <YOUR VPC-ID>\ cifar10-deep-learning
Training with GPU
docker-compose -f docker-compose-gpu.yml up -d
Training with CPU
docker-compose -f docker-compose-cpu.yml up -d
Please check the article for more A Practical example of Training a Neural Network in the AWS cloud with Dockerfor more details.