README.md 854 Bytes
Newer Older
zhanggezhong's avatar
zhanggezhong committed
1
2
3
4
5
6
# <div align="center"><strong>FastMoe</strong></div>
## 简介
一个易于使用和高效的系统,支持PyTorch的混合专家(MoE)模型。
## 安装
  方式1:
  下载whl包,通过pip3 install 安装
zhanggezhong's avatar
zhanggezhong committed
7
  
zhanggezhong's avatar
zhanggezhong committed
8
9
10
11
12
13
14
15
  方式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
zhanggezhong's avatar
zhanggezhong committed
16
17
18
  python3 setup.py build
  python3 setup.py install 
  python3 setup.py bdist_wheel # 该指令用于编译whl包,执行该指令时不必执行前两个指令
zhanggezhong's avatar
zhanggezhong committed
19
20
  ```
## 验证安装
Rick Ho's avatar
fmoefy  
Rick Ho committed
21
```
zhanggezhong's avatar
zhanggezhong committed
22
23
24
25
26
27
28
pip3 list | grep fastmoe
python3
import fmoe 
fmoe.__version__
#返回版本号
fmoe.__dcu_version
#返回版本号和dcu版本号
Rick Ho's avatar
Rick Ho committed
29
```
zhanggezhong's avatar
zhanggezhong committed
30
## 测试
Rick Ho's avatar
Rick Ho committed
31
```
zhanggezhong's avatar
zhanggezhong committed
32
cd fastmoe/tests
zhanggezhong's avatar
zhanggezhong committed
33
pytest vs
Rick Ho's avatar
Rick Ho committed
34

Rick Ho's avatar
Rick Ho committed
35
```