pytorch-aws

Example code for PyTorch on AWS made easy AWS bits are done through ncluster library.

Tested with PyTorch 1.1/Python 3.6

conda install -c pytorch pytorch torchvision -y
git clone https://github.com/cybertronai/pytorch-aws
cd pytorch-aws
pip install -r requirements.txt
python mnist.py