Commit b42ba980 authored by qianyj's avatar qianyj
Browse files

Adapt README.md

parent 2e782c25
......@@ -16,14 +16,6 @@ NNI 管理自动机器学习 (AutoML) 的 Experiment,**调度运行** 由调
### **安装方式**
#### **使用pip方式安装**
nni whl包下载目录:[光合社区/资源工具/AI生态包/nni/dtk23.04](https://cancon.hpccube.com:65024/4/main/nni/dtk23.04),根据python版本选择对应nni的whl包。
```bash
pip3 install nni*(下载的nni的whl包)
```
#### **使用源码编译方式安装**
编译之前,需要先安装对应版本python,安装相应的三方包依赖项,并配置DTK环境变量(以Centos7.x为例)。
......@@ -42,21 +34,28 @@ pip3 install protobuf==3.20
pip3 install wheel
pip3 install jupyter
pip3 install jupyterlab==3.0.9
pip3 install colorama
pip3 install cloudpickle
pip3 install json_tricks
pip3 install 'typeguard<3'
pip3 install astor
pip3 install schema
python3 -m pip install --upgrade pip setuptools
# 下载DTK并配置环境变量
# DTK tar包下载目录:光合社区/资源工具/DCU Toolkit/DTK23.04(https://cancon.hpccube.com:65024/1/main/DTK-23.04),根据系统选择对应DTK的tar包,并解压至/opt目录。
export ROCM_PATH=/opt/dtk-23.04
source /opt/dtk-23.04/env.sh
# 下载DTK及hyhal并配置环境变量
# DTK tar包下载目录:光合社区/资源工具/DCU Toolkit/DTK23.10(https://cancon.hpccube.com:65024/1/main/DTK-23.10),根据系统选择对应DTK的tar包,并解压至/opt目录。
# hyhal tar包下载目录: 光合社区/资源工具DCU Toolkit/DTK23.10(https://cancon.hpccube.com:65024/1/main/DTK-23.10),并解压至/opt目>录。
export ROCM_PATH=/opt/dtk-23.10
source /opt/dtk-23.10/env.sh
```
编译nni
```bash
# 下载源码
git clone -b dtk23.04-v2.9 http://developer.hpccube.com/codes/aicomponent/nni.git
git clone -b dtk23.10-v2.10 http://developer.hpccube.com/codes/aicomponent/nni.git
cd nni
export NNI_RELEASE=2.9
export NNI_RELEASE=2.10
python3 setup.py clean --all
python3 setup.py build_ts
python3 setup.py bdist_wheel -p manylinux1_x86_64
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment