安装#

你可以使用 pip 安装 Mars:

pip install pymars

创建默认 session#

Mars 需要创建一个 Session 来执行代码。你可以使用 mars.new_session() 在本机创建一个 Session:

import mars
mars.new_session()

在 Mars 对象比如 tensor 上调用 execute() 方法会触发计算。

a = mt.random.rand(10, 10)
a.dot(a.T).execute()

如果默认 Session 没有被创建,Mars 将会在执行代码时创建一个新的 Session,你将在创建 Session 时看到相关的消息。

当你在本地 Session 上完成计算后,可以使用 mars.stop_server() 停止该 Session。

mars.stop_server()