高斯混合模型的优化(和可选的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()的结果匹配,才能获取 主机字符串。

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
Java程序运行时错误   JavaAndroidStudio:与往常一样,四舍五入到next.5或.0   apache使用Java以表单数据形式上载文件   带矢量的java Freeflight相机如何正确旋转?   java如何以编程方式检索有关当前项目的语言、操作系统、体系结构等信息   java Twitter4J tweet实体?   java PdfBox编码异常   java在拖动未装饰的舞台时,如何强制光标停留在窗口上   JavaSpring注释扫描优化   java无法通过IntelliJ Idea在tomcat上运行服务   java在生命周期中如何拦截请求?   java中的数组返回错误