# Colossal-AI

## 安装
Colossal-AI支持
+ Python 3.7.
+ Python 3.8.
+ Python 3.9.

### 使用pip安装
colossalai whl包下载目录：[https://cancon.hpccube.com:65024/4/main/colossalai/dtk23.04](https://cancon.hpccube.com:65024/4/main/colossalai/dtk23.04)
根据对应的pytorch版本和python版本，下载对应colossalai的whl包

```shell
pip install colossalai* (下载的colossalai的whl包)
```

### 使用源码安装

#### 编译环境准备
pytorch whl包下载目录：[https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04](https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04)
根据python版本,下载对应pytorch的whl包
```shell
pip install torch* (下载的torch的whl包)
```

```shell
pip install setuptools wheel
```

#### 编译安装
```shell
git clone -b dtk-23.04_colossalai0.1.13 https://developer.hpccube.com/codes/aicomponent/colossalai
cd colossalai
export COLOSSALAI_BUILD_VERSION=abix.dtkxxx
python setup.py bdist_wheel 
pip install dist/colossalai*
```

## Note
+ 若使用 pip install 下载安装过慢，可添加源：-i https://pypi.tuna.tsinghua.edu.cn/simple/
+ COLOSSALAI_BUILD_VERSION为编译的版本号设置，版本号为0.1.13+gitxxx.abix.dtkxxx 
gitxxx：为代码自动获取；abi0：使用devtools的gcc编译；abi1：使用非devtools的gcc编译; dtkxxx为dtk的版本号：例如：dtk2304