megengine.set_default_device¶
- set_default_device(device='xpux')[源代码]¶
设置默认的计算节点。
- 参数
device (
str) – default device type.
注解
The type can be ‘cpu0’, ‘cpu1’, etc., or ‘gpu0’, ‘gpu1’, etc., to specify the particular CPU or GPU to use.
‘cpux’ and ‘gpux’ can also be used to specify any number of CPU or GPU devices.
The default value is ‘xpux’ to specify any device available.
The priority of using GPU is higher when both GPU and CPU are available.
‘multithread’ device type is avaliable when inference, which implements multi-threading parallelism at the operator level. For example, ‘multithread4’ will compute with 4 threads.
It can also be set by environment variable
MGE_DEFAULT_DEVICE.