在 Linux 和 Mac 下安装¶
安装¶
在 Linux 和 macOS 上安装，遵循以下相同的说明。
通过 pip 命令安装 NNI¶
先决条件：
python 64-bit >= 3.6
python3 -m pip install --upgrade nni
通过源代码安装 NNI¶
如果对某个或最新版本的代码感兴趣，可通过源代码安装 NNI。
先决条件：
python 64-bit >=3.6,git
git clone -b v2.6 https://github.com/Microsoft/nni.git
cd nni
python3 -m pip install -U -r dependencies/setup.txt
python3 -m pip install -r dependencies/develop.txt
python3 setup.py develop
从 NNI 源代码构建 Wheel 包¶
上一节介绍了如何在 开发模式 下安装NNI。 如果要执行持久安装，建议您构建自己的 wheel 软件包并从wheel 安装。
git clone -b v2.6 https://github.com/Microsoft/nni.git
cd nni
export NNI_RELEASE=2.6
python3 -m pip install -U -r dependencies/setup.txt
python3 -m pip install -r dependencies/develop.txt
python3 setup.py clean --all
python3 setup.py build_ts
python3 setup.py bdist_wheel -p manylinux1_x86_64
python3 -m pip install dist/nni-2.6-py3-none-manylinux1_x86_64.whl
在 Docker 映像中使用 NNI¶
也可将 NNI 安装到 docker 映像中。 参考 这里 来生成 NNI 的 docker 映像。 也可通过此命令从 Docker Hub 中直接拉取 NNI 的映像
docker pull msranni/nni:latest。
验证安装¶
通过克隆源代码下载示例。
git clone -b v2.6 https://github.com/Microsoft/nni.git
运行 MNIST 示例。
nnictl create --config nni/examples/trials/mnist-pytorch/config.yml
在命令行中等待输出
INFO: Successfully started experiment!。 此消息表明实验已成功启动。 通过命令行输出的 Web UI url 来访问 Experiment 的界面。
INFO: Starting restful server...
INFO: Successfully started Restful server!
INFO: Setting local config...
INFO: Successfully set local config!
INFO: Starting experiment...
INFO: Successfully started experiment!
-----------------------------------------------------------------------
The experiment id is egchD4qy
The Web UI urls are: http://223.255.255.1:8080 http://127.0.0.1:8080
-----------------------------------------------------------------------
You can use these commands to get more information about the experiment
-----------------------------------------------------------------------
commands description
1. nnictl experiment show show the information of experiments
2. nnictl trial ls list all of trial jobs
3. nnictl top monitor the status of running experiments
4. nnictl log stderr show stderr log content
5. nnictl log stdout show stdout log content
6. nnictl stop stop an experiment
7. nnictl trial kill kill a trial job by id
8. nnictl --help get help information about nnictl
-----------------------------------------------------------------------
在浏览器中打开
Web UI url，可看到下图的实验详细信息，以及所有的尝试任务。 查看 这里 的更多页面。
系统需求¶
由于程序变更，NNI 的最低配置会有所更改。
Linux¶
推荐配置 |
最低配置 |
|
|---|---|---|
操作系统 |
Ubuntu 16.04 或以上版本 |
|
CPU |
Intel® Core™ i5 或 AMD Phenom™ II X3 或更高配置 |
Intel® Core™ i3 或 AMD Phenom™ X3 8650 |
GPU |
NVIDIA® GeForce® GTX 660 或更高配置 |
NVIDIA® GeForce® GTX 460 |
内存 |
6 GB |
4 GB |
存储 |
30 GB 可用的磁盘空间 |
|
网络 |
宽带连接 |
|
分辨率 |
1024 x 768 以上 |
macOS¶
推荐配置 |
最低配置 |
|
|---|---|---|
操作系统 |
macOS 10.14.1 或更高版本 |
|
CPU |
Intel® Core™ i7-4770 或更高 |
Intel® Core™ i5-760 或更高 |
GPU |
AMD Radeon™ R9 M395X 或更高 |
NVIDIA® GeForce® GT 750M 或 AMD Radeon™ R9 M290 或更高 |
内存 |
8 GB |
4 GB |
存储 |
70GB 可用空间 SSD 硬盘 |
70GB 可用空间及 7200 RPM 硬盘 |
网络 |
宽带连接 |
|
分辨率 |
1024 x 768 以上 |