megengine.set_default_device¶
- set_default_device(device='xpux')[源代码]¶
设置默认的计算节点。
- 参数
device (
str
) – 默认的设备类型
注解
这个类型可以是’cpu0’, ‘cpu1’等, 或者 ‘gpu0’, ‘gpu1’等,用来指定要使用的特定CPU或者GPU。
‘cpux’ 和 ‘gpux’也可以用来指定任意数量的CPU或者GPU设备。
默认值是’xpux’,用来指代任意类型的可用设备。
当GPU和CPU都可用时,使用GPU的优化级更高。
‘多线程’设备类型在推理时是可用的,它在算子层面实现了多线程并行。比如,’multithread4’会开4个线程进行计算。
也可以通过环境变量``MGE_DEFAULT_DEVICE``来设置。