khanhptnk/mnist-pytorch

Basic MNIST Example

This code is a copy from:

https://github.com/pytorch/examples/tree/master/mnist

with slight modifications that fix problems with downloading the data.

pip install -r requirements.txt
python main.py
# CUDA_VISIBLE_DEVICES=2 python main.py  # to specify GPU id to ex. 2