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``来设置。