PaiMode.md 6.08 KB
Newer Older
1
2
**Run an Experiment on OpenPAI**
===
Chi Song's avatar
Chi Song committed
3
NNI supports running an experiment on [OpenPAI](https://github.com/Microsoft/pai) (aka pai), called pai mode. Before starting to use NNI pai mode, you should have an account to access an [OpenPAI](https://github.com/Microsoft/pai) cluster. See [here](https://github.com/Microsoft/pai#how-to-deploy) if you don't have any OpenPAI account and want to deploy an OpenPAI cluster. In pai mode, your trial program will run in pai's container created by Docker.
4
5

## Setup environment
xuehui's avatar
xuehui committed
6
Install NNI, follow the install guide [here](../Tutorial/QuickStart.md).
7
8

## Run an experiment
9
Use `examples/trials/mnist-annotation` as an example. The NNI config YAML file's content is like:
Chi Song's avatar
Chi Song committed
10

Yan Ni's avatar
Yan Ni committed
11
```yaml
12
13
14
15
16
17
18
19
20
21
authorName: your_name
experimentName: auto_mnist
# how many trials could be concurrently running
trialConcurrency: 2
# maximum experiment running duration
maxExecDuration: 3h
# empty means never stop
maxTrialNum: 100
# choice: local, remote, pai
trainingServicePlatform: pai
LongzeSong's avatar
LongzeSong committed
22
23
# search space file
searchSpacePath: search_space.json
24
# choice: true, false
25
26
27
28
29
30
31
32
33
34
35
useAnnotation: true
tuner:
  builtinTunerName: TPE
  classArgs:
    optimize_mode: maximize
trial:
  command: python3 mnist.py
  codeDir: ~/nni/examples/trials/mnist-annotation
  gpuNum: 0
  cpuNum: 1
  memoryMB: 8196
36
  image: msranni/nni:latest
37
38
39
  virtualCluster: default
  nniManagerNFSMountPath: /home/user/mnt
  containerNFSMountPath: /mnt/data/user
40
41
42
# Configuration to access OpenPAI Cluster
paiConfig:
  userName: your_pai_nni_user
43
  token: your_pai_token
44
45
  host: 10.1.1.1
```
Chi Song's avatar
Chi Song committed
46

47
Note: You should set `trainingServicePlatform: pai` in NNI config YAML file if you want to start experiment in pai mode.
48

xuehui's avatar
xuehui committed
49
Compared with [LocalMode](LocalMode.md) and [RemoteMachineMode](RemoteMachineMode.md), trial configuration in pai mode have these additional keys:
50
51
52
53
54
* cpuNum
    * Required key. Should be positive number based on your trial program's CPU  requirement
* memoryMB
    * Required key. Should be positive number based on your trial program's memory requirement
* image
55
    * Required key. In pai mode, your trial program will be scheduled by OpenPAI to run in [Docker container](https://www.docker.com/). This key is used to specify the Docker image used to create the container in which your trial will run.
Chi Song's avatar
Chi Song committed
56
    * We already build a docker image [nnimsra/nni](https://hub.docker.com/r/msranni/nni/) on [Docker Hub](https://hub.docker.com/). It contains NNI python packages, Node modules and javascript artifact files required to start experiment, and all of NNI dependencies. The docker file used to build this image can be found at [here](https://github.com/Microsoft/nni/tree/master/deployment/docker/Dockerfile). You can either use this image directly in your config file, or build your own image based on it.
57
* virtualCluster
58
    * Optional key. Set the virtualCluster of OpenPAI. If omitted, the job will run on default virtual cluster.
59
60
61
62
63
64
* nniManagerNFSMountPath
    * Required key. Set the mount path in your nniManager machine.
* containerNFSMountPath
    * Required key. Set the mount path in your container used in PAI.
* paiStoragePlugin
    * Required key. Set the storage plugin name used in PAI.
65

66

67
Once complete to fill NNI experiment config file and save (for example, save as exp_pai.yml), then run the following command
68
```
69
nnictl create --config exp_pai.yml
70
```
71
to start the experiment in pai mode. NNI will create OpenPAI job for each trial, and the job name format is something like `nni_exp_{experiment_id}_trial_{trial_id}`.
72
You can see jobs created by NNI in the OpenPAI cluster's web portal, like:
xuehui's avatar
xuehui committed
73
![](../../img/nni_pai_joblist.jpg)
74

75
Notice: In pai mode, NNIManager will start a rest server and listen on a port which is your NNI WebUI's port plus 1. For example, if your WebUI port is `8080`, the rest server will listen on `8081`, to receive metrics from trial job running in Kubernetes. So you should `enable 8081` TCP port in your firewall rule to allow incoming traffic.
76

77
Once a trial job is completed, you can goto NNI WebUI's overview page (like http://localhost:8080/oview) to check trial's information.
78
79

Expand a trial information in trial list view, click the logPath link like:
xuehui's avatar
xuehui committed
80
![](../../img/nni_webui_joblist.jpg)
81
82

And you will be redirected to HDFS web portal to browse the output files of that trial in HDFS:
xuehui's avatar
xuehui committed
83
![](../../img/nni_trial_hdfs_output.jpg)
84
85
86

You can see there're three fils in output folder: stderr, stdout, and trial.log

87
## data management
88
Befour using NNI to start your experiment, users should set the corresponding mount data path in your nniManager machine. PAI has their own storage(NFS, AzureBlob ...), and the storage will used in PAI will be mounted to the container when it start a job. Users should set the PAI storage type by `paiStoragePlugin` field to choose a storage in PAI. Then users should mount the storage to their nniManager machine, and set the `nniManagerNFSMountPath` field in configuration file, NNI will generate bash files and copy data in `codeDir` to the `nniManagerNFSMountPath` folder, then NNI will start a trial job. The data in `nniManagerNFSMountPath` will be sync to PAI storage, and will be mounted to PAI's container. The data path in container is set in `containerNFSMountPath`, NNI will enter this folder first, and then run scripts to start a trial job. 
89
90

## version check
91
92
NNI support version check feature in since version 0.6. It is a policy to insure the version of NNIManager is consistent with trialKeeper, and avoid errors caused by version incompatibility.
Check policy:
93
1. NNIManager before v0.6 could run any version of trialKeeper, trialKeeper support backward compatibility.
94
95
96
97
2. Since version 0.6, NNIManager version should keep same with triakKeeper version. For example, if NNIManager version is 0.6, trialKeeper version should be 0.6 too.
3. Note that the version check feature only check first two digits of version.For example, NNIManager v0.6.1 could use trialKeeper v0.6 or trialKeeper v0.6.2, but could not use trialKeeper v0.5.1 or trialKeeper v0.7.

If you could not run your experiment and want to know if it is caused by version check, you could check your webUI, and there will be an error message about version check.
98
![](../../img/version_check.png)