/docker-cifar

Serving a PyTorch machine learning model within a Docker container using GPU inference

Primary LanguagePython

PyTorch machine learning GPU inference using Docker and Flask

png

This project illustrates how to serve a PyTorch model as a REST API using a simple CIFAR model for illustration from the PyTorch CIFAR tutorial.

This is done using Flask, Docker and PyTorch.

Blog post on paepper.com/blog

For full details about the approach and how to run it yourself, see my blog post covering the details