Commit 6ef178d0 authored by zhuwenwen's avatar zhuwenwen
Browse files

modify readme

parent 43ff1d4f
...@@ -29,12 +29,31 @@ pip install setuptools wheel ...@@ -29,12 +29,31 @@ pip install setuptools wheel
#### 编译安装 #### 编译安装
```shell ```shell
git clone -b dtk-23.04_colossalai0.1.13 https://developer.hpccube.com/codes/aicomponent/colossalai git clone https://developer.hpccube.com/codes/aicomponent/colossalai # 根据需要的分支进行切换
cd colossalai cd colossalai
1、 编译whl包并安装
python setup.py bdist_wheel python setup.py bdist_wheel
pip install dist/colossalai* pip install dist/colossalai*
2、 直接编译安装
python setup.py install
``` ```
## 安装包命名
例: colossalai-0.1.13+git43ff1d4.abi0.dtk2304.torch1.10-cp37-cp37m-manylinux2014_x86_64.whl
- colossalai: 安装包名称;
- 0.1.13: 安装包版本号;
- git43ff1d4: git号;
- abi0: 对应centos和rocky多系统的标识,取值为abi0(centos7),abi1(rocky8),可使用命令行动态查询获取辅助判断(echo '#include <string>' | gcc -x c++ -E -dM - | fgrep _GLIBCXX_USE_CXX11_ABI);
- dtk2304: 对应dtk大版本号,命令行动态查询取值(/opt/dtk-23.04/.info/rocm_version);
- torch1.10: 对应构建的torch版本号,命令行动态查询取值(torch.\_\_version__);
- cp37-cp37m: 对应python版本号;
- manylinux2014_x86_64: 系统架构;
## 版本号查询
- colossalai.\_\_version__:与官方版本同步,查询该安装包的版本号,例如0.1.13;
## Note ## Note
+ 若使用 pip install 下载安装过慢,可添加源:-i https://pypi.tuna.tsinghua.edu.cn/simple/ + 若使用 pip install 下载安装过慢,可添加源:-i https://pypi.tuna.tsinghua.edu.cn/simple/
......
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