未验证 提交 bca23057 编写于 作者: K kouml 提交者: GitHub

Add requirements for cifar10 (#79)

* add install requirements command line

* add pillow library to fix version

* modify to uppercase
Co-authored-by: NJeff Rasley <jerasley@microsoft.com>
上级 807480a0
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册