Deploying a basic mnist digit classifier using pytorch and flask
git clone https://github.com/rtharungowda/mnist-flask-pytorch
Then
cd mnist-flsk-pytorch
It is suggested to create a new virtual env
Install venv pip3 install python3-venv
Create a new venv in the folder mnist-flask-pytorch
python3 -m venv <venv>
. <venv>/bin/activate
Then install dependencies, PS Note
pip3 install -r requirements.txt
Pytorch version (i.e, is torch and torchvision) support only cpu , as for deployment a gpu is always not available and the task is light weight
Hence comment out torch
torchvision
and the first line and use the commands given in Pytorch website
depeding on system and availability of cuda