Tutorial on how to deploy Deep Learning models on GPU enabled Kubernetes cluster
|size (curr.)||6899 kB|
This repo is no longer actively maintained, please see newer version available using Azure Machine Learning here.
Authors: Mathew Salvaris and Fidan Boylu Uz
Deploy Deep Learning CNN on Kubernetes Cluster with GPUs
In this repository there are a number of tutorials in Jupyter notebooks that have step-by-step instructions on how to deploy a pretrained deep learning model on a GPU enabled Kubernetes cluster. The tutorials cover how to deploy models from the following deep learning frameworks:
For each framework, we go through the following steps:
- Model development where we load the pretrained model and test it by using it to score images
- Developing the interface our Flask app will use to load and call the model
- Building the Docker Image with our Flask REST API and model
- Testing our Docker image before deployment
- Creating our Kubernetes cluster and deploying our application to it
- Testing the deployed model
- Testing the throughput of our model
- Cleaning up resources
The application we will develop is a simple image classification service, where we will submit an image and get back what class the image belongs to. The application flow for the deep learning model is as follows:
- The client sends a HTTP POST request with the encoded image data.
- The Flask app extracts the image from the request.
- The image is then appropriately preprocessed and sent to the model for scoring.
- The scoring result is then piped into a JSON object and returned to the client.
If you already have a Docker image that you would like to deploy you can skip the first four notebooks.
NOTE: The tutorial goes through step by step how to deploy a deep learning model on Azure; it does not include enterprise best practices such as securing the endpoints and setting up remote logging etc.
Deploying with GPUS: For a detailed comparison of the deployments of various deep learning models, see the blog post here which provides evidence that, at least in the scenarios tested, GPUs provide better throughput and stability at a lower cost.
- Linux(Ubuntu). The tutorial was developed on an Azure Linux DSVM
- Docker installed. NOTE: Even with docker installed you may need to set it up so that you don’t require sudo to execute docker commands see “Manage Docker as a non-root user”
- Dockerhub account
- Port 9999 open: Jupyter notebook will use port 9999 so please ensure that it is open. For instructions on how to do that on Azure see here
- Clone the repo:
git clone <repo web URL>
- Login to Docker with your username and password.
- Go to the framework folder you would like to run the notebooks for.
- Create a conda environment:
conda env create -f environment.yml
- Activate the environment:
source activate <environment name>
- Start the first notebook and make sure the kernel corresponding to the above environment is selected.
After following the setup instructions above, run the Jupyter notebooks in order. The same basic steps are followed for each deep learning framework.
To remove the conda environment created see here. The last Jupyter notebook within each folder also gives details on deleting Azure resources associated with this repo.
This project welcomes contributions and suggestions. Most contributions require you to agree to a Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us the rights to use your contribution. For details, visit https://cla.microsoft.com.
When you submit a pull request, a CLA-bot will automatically determine whether you need to provide a CLA and decorate the PR appropriately (e.g., label, comment). Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA.