Files

201_mnist

Name		Name	Last commit message	Last commit date
parent directory ..
MNIST.ipynb		MNIST.ipynb
README.md		README.md
main.py		main.py
requirements.txt		requirements.txt

README.md

Basic MNIST Example

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