Skip to content

mpaepper/docker-cifar

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Docker based GPU inference of machine learning models

Topics

Resources

Stars

Watchers

Forks