昇腾社区首页
中文
注册
开发者
下载

简介

GeApi类提供全局GE初始化和终结功能。

GE初始化调用示例:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
from ge.ge_global import GeApi

# 初始化 GE
config = {
    "ge.execDeviceId": "0",
    "ge.graphRunMode": "0"
}
GeApi.ge_initialize(config)

# ... 执行操作 ...

# 终结 GE
GeApi.ge_finalize()