diff --git a/docs/tutorials/CIFAR-10.md b/docs/tutorials/CIFAR-10.md index 8889d91ff544..6dda5dcab4af 100755 --- a/docs/tutorials/CIFAR-10.md +++ b/docs/tutorials/CIFAR-10.md @@ -18,6 +18,12 @@ Original model code from [CIFAR-10 Tutorial](https://github.com/pytorch/tutorial git submodule update --init --recursive ``` +To install requirements for CIFAR-10: +``` +cd DeepSpeedExamples/cifar +pip install -r requirements.txt +``` + Run `python cifar10_tutorial.py`, it downloads the training data set at first run. ```less Downloading https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz to ./data/cifar-10-python.tar.gz diff --git a/requirements.txt b/requirements.txt index 4d92334a9111..dd2c5fdf478f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,6 @@ torch==1.2 torchvision==0.4.0 +pillow==6.2.2 tqdm psutil tensorboardX==1.8