环境准备

我们可以在CPU的云主机上为MNIST的训练做准备,该云主机需要满足以下条件:

  1. 需要是Linux或类Linux环境
  2. 安装docker,建议使用docker-ce,且版本 > 10.0
  3. 该主机可以访问uhub.service.auto-ai.com.cn 或 uhub.auto-ai.com.cn

本Tutorial将以云知芯 普通云主机为范例操作(你也可以使用自己的主机或其他系统)

创建云主机

我们根据ubuntu云主机申请操作申请一台2核4G的CPU云主机作为操作平台。

安装docker

1.设置官方docker软件包源

sudo apt-get -y install \
  apt-transport-https \
  ca-certificates \
  curl

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

sudo add-apt-repository \
       "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
       $(lsb_release -cs) \
       stable"

sudo apt-get update

2.安装docker-ce

sudo apt-get -y install docker-ce

3.测试docker安装

sudo docker run hello-world

安装UAI SDK

我们需要从github上面下载uai-sdk:

$ cd ~
$ git clone https://github.com/ucloud/uai-sdk.git
$ cd uai-sdk
$ sudo python setup.py install

准备操作环境

我们在/data/目录下创建一个mnist目录来执行操作:

$ cd /data/
$ mkdir mnist/

下载数据

我们可以从http://yann.lecun.com/exdb/mnist/下载Mnist的训练数据和测试数据,你也可以从https://github.com/ucloud/uai-sdk/tree/master/examples/tensorflow/train/mnist_summary_1.1/data获取,我们将数据放入/data/mnist/data/路径:

$ mkdir /data/mnist/data/
$ cd /data/mnist/data/
$ wget http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz
$ wget http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz
$ wget http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz
$ wget http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz
$ ls
train-images-idx3-ubyte.gz  train-labels-idx1-ubyte.gz  t10k-images-idx3-ubyte.gz  t10k-labels-idx1-ubyte.gz

下载代码

我们可以直接从https://github.com/ucloud/uai-sdk/tree/master/examples/tensorflow/train/mnist_summary_1.1获取代码,并放入/data/mnist/code/路径。由于之前我们已经从github下载的uai-sdk,我们只需要去examples下面拷贝 tensorflow mnist例子的代码至/data/mnist下面即可:

$ cp -fr ~/uai-sdk/examples/tensorflow/train/mnist_summary_1.1/code/ /data/mnist/

环境Ready

至此,我们已经准备好了

  1. 基础环境:docker
  2. UAI-SDK环境:uai-sdk
  3. 代码和数据
|_ data
|  |_ mnist
|  |  |_ code
|  |  |  |_ mnist_summary.py
|  |  |_ data
|  |  |  |_ train-images-idx3-ubyte.gz
|  |  |  |_ train-labels-idx1-ubyte.gz
|  |  |  |_ t10k-images-idx3-ubyte.gz
|  |  |  |_ t10k-labels-idx1-ubyte.gz