megengine.device.
DeviceType
基类:pybind11_builtins.pybind11_object
pybind11_builtins.pybind11_object
成员:
UNSPEC
CUDA
CPU
MULTITHREAD
MAX_DEVICE_ID
__eq__
__getstate__
__hash__
__init__
__int__
__members__
__ne__
__repr__
__setstate__
name
get_default_device
查询默认的计算结点。
返回被 set_default_device() 设置的值。
set_default_device()
str
get_device_count
获取当前系统中的设备数量。
device_type (str) – 设备类型,’gpu’ 和 ‘cpu’ 中的一个
int
get_mem_status_bytes
以字节为单位获取计算设备上的总内存和可用内存。
is_cuda_available
cuda设备是否在当前机器可用。
bool
set_default_device
设置默认的计算节点。
device (str) – 默认设备类型。类型可以是 ‘cpu0’,’cpu1’ 等,或 ‘gpu0’,’gpu1’ 等,用以指定要使用的特定 cpu 或 gpu。’cpux’ 和’gpux’ 也可以用于指定任意数量的 cpu 或 gpu 设备。’multithread’ 设备类型在推理时可用,它实现了操作员级别的多线程并行性。例如,’multithread4’ 将使用4个线程进行计算。默认值是 ‘xpux’,以指定任何可用的设备。当gpu和cpu都可用时,优先使用gpu。也可以通过以下方式设置环境变量`MGE_DEFAULT_DEVICE`。
set_prealloc_config
对原始设备的内存分配器指定预先分配的方案。
alignment (int) – 指定对齐的字节数。
min_req (int) – 最小的需要的字节数。
max_overhead (int) – 为满足需要的字节数花费的开销上限(单位:字节)。
growth_factor – 需要的大小和目前分配的大小的比值
device_type – 设备类别
megengine.logger.
MegBrainLogFormatter
基类:megengine.logger.MegEngineLogFormatter
megengine.logger.MegEngineLogFormatter
date
MegEngineLogFormatter
基类:logging.Formatter
logging.Formatter
date_full
format
将指定的记录格式化为文本。
record的属性字典用作字符串格式化操作中的操作数,该操作产生返回的字符串。在格式化字典之前,需要执行几个准备步骤。使用LogRecord.getMessage()计算record的消息属性。如果格式化字符串使用时间(通过调用usesTime()来确定),则将调用formatTime()来格式化事件时间。如果存在异常信息,则使用formatException()将其格式化并附加到消息中。
log_fout
max_lines
msg
enable_debug_log
设置日志记录级别(logging level)以调试所有组件。
get_logger
按照给定名称获取megengine日志。
replace_mgb_log_level
替换block中megbrain的日志级别,并在退出后恢复。
level (int e.g. logging.INFO) – 新的日志级别
set_log_file
设置日志的输出文件。
fout (str or file-like) – 支持写操作和flush操作的文件对象,或文件名的字符串
mode (str) – 如果 fout 是一个字符串,则为指定打开日志文件的模式
set_log_level
设置默认的日志记录级别。
level (int e.g. logging.INFO) – python的 logging 模块给出的日志记录级别
logging
update_existing – 是否更新现有的日志记录
set_mgb_log_level
设置megbrain的日志级别
初始的日志级别
megengine.serialization.
dmap
基类:object
object
load
从文件中加载用save()保存的对象。
f (text file object) – 文件名字符串或一个需要加载的文件对象。
map_location (str, dict or a function specifying the map rules) – 默认值:None
注解
map_location 定义了设备的映射。用法请参考示例。
pickle_module – 默认值: pickle
pickle
如果你要调用 mge.set_default_device(),请在调用 mge.load() 前完成。
mge.set_default_device()
mge.load()
例如:
save
把对象保存成磁盘文件。
obj (object) – 要保存的对象。只支持 module 和 state_dict 。
module
state_dict
f (text file object) – 文件名字符串或一个需要保存的文件对象。
pickle_protocol – 默认值: pickle.HIGHEST_PROTOCOL
pickle.HIGHEST_PROTOCOL
megengine.tensor.
Parameter
基类:megengine.tensor.Tensor
megengine.tensor.Tensor
一类表示模块参数的张量。
Tensor
基类:Tensor, megengine.core.tensor.array_method.ArrayMethodMixin
megengine.core.tensor.array_method.ArrayMethodMixin
__getnewargs__
__getstate__ 会在 pickle 序列化或深拷贝时被调用
dmap_callback
grad
q_dict
requires_grad
reset_zero
1.0 版后已移除.
请使用 *= 0
set_value
在1.0版本后不需要重复使用一个存在的张量
shape
to
apply
tensor
megengine.tensor.Tensor 的别名