高斯混合模型的优化(和可选的gpu增强)拟合
dpmix的Python项目详细描述
dpmix是一个库,用于了解 正态分布的dirichlet与hierarchical-dirichlet混合 以截断的断棍表示。
要求
- 努比
- scipy
- 赛顿
- 皮库达
- 赛尔玛
- 赛兰德
- scikits.cuda
- G自定义
- MPI4PY
安装和测试
通过安装
python setup.py install
要测试,请运行“测试”子文件夹中的脚本。
用法
查看类docstrings以获取更多信息。
MPI
多GPU设备是使用MPI开发的。因此, 使用多台机器是可能的。但是,请注意 必须以相同的方式配置计算机。(Python)
在多台计算机上运行代码需要mpiexec:
mpiexec -hostfile my_hosts -np 1 python tests/test_dpmix.py --gpu MPI
其中my_宿主文件看起来像
host1 slots=3 host2 slots=2
我假设python的主实例在host1上运行 host1和host2各有2个gpu。注意,需要有一个额外的插槽 为主机1上的主机保留。此外,我们需要具体说明 要在每个主机上使用的设备。类构造函数中的gpu参数 必须是类似于
gpu={'host1': [0,1], 'host2': [0,1]}
密钥必须与调用os.uname()的结果匹配,才能获取 主机字符串。