#
FastMoe
## 简介
一个易于使用和高效的系统,支持PyTorch的混合专家(MoE)模型。
## 安装
方式1:
下载whl包,通过pip3 install 安装
方式2,源码编译安装,该方式需要安装torch及fastpt工具包
```
pip3 install dm-tree
pip3 install pytest
pip3 install wheel
git clone https://developer.hpccube.com/codes/OpenDAS/fastmoe.git
git checkout * #切换到相应分支
cd fastmoe
python3 setup.py build
python3 setup.py install
python3 setup.py bdist_wheel # 该指令用于编译whl包,执行该指令时不必执行前两个指令
```
## 验证安装
```
pip3 list | grep fastmoe
python3
import fmoe
fmoe.__version__
#返回版本号
fmoe.__dcu_version
#返回版本号和dcu版本号
```
## 测试
```
cd fastmoe/tests
pytest vs
```