#
Uni-Core
## 简介
Uni-Core 专为快速创建高性能 PyTorch 模型而构建,尤其是基于 Transfromer 的模型。详细信息可参考README_ORIGIN.md
## 安装
源码编译安装,该方式需要安装torch及fastpt工具包;注意使用fastpt包进行源码编译安装时,要严格匹配fastpt、torch、dtk之间的版本号,例如基于dtk2504编译,则fastpt、torch都必须是dtk2504的包,其中fastpt与torch对应的版本号关系为
| | fastpt版本 | torch版本 | DTK版本 |
| - | -------- | ------- | ------------ |
| 1 | 2.0.1+das.dtk2504 | v2.4.1 | dtk2504|
| 1 | 2.1.0+das.dtk2504 | v2.5.1 | dtk2504|
| 1 | 2.0.1+das.dtk25041 | v2.4.1 | dtk25041|
| 1 | 2.1.0+das.dtk25041 | v2.5.1 | dtk25041|
### 编译流程
```
pip3 install wandb
pip3 install -r requirements.txt
pip3 install fastpt-2.0.1+das.dtk2504-py3-none-any.whl #以torch2.4.1,dtk2504为例
git clone https://developer.sourcefind.cn/codes/OpenDAS/Uni-Core.git
cd Uni-Core
git checkout v0.0.1-fastpt #切换到相应分支
source /usr/local/bin/fastpt -c
python3 setup.py bdist_wheel
```
## 验证安装
```
pip3 list | grep unicore
python3
import unicore
unicore.__version__
#返回版本号
```
## 测试
```
source /usr/local/bin/fastpt -e
cd tests
pytest vs
```