megengine.set_default_device¶
-
set_default_device(device='xpux')[源代码]¶ 设置默认的计算节点。
- 参数
device (
str) – 默认设备类型。类型可以是 ‘cpu0’，’cpu1’ 等，或 ‘gpu0’，’gpu1’ 等，用以指定要使用的特定 cpu 或 gpu。’cpux’ 和’gpux’ 也可以用于指定任意数量的 cpu 或 gpu 设备。’multithread’ 设备类型在推理时可用，它实现了操作员级别的多线程并行性。例如，’multithread4’ 将使用4个线程进行计算。默认值是 ‘xpux’，以指定任何可用的设备。当gpu和cpu都可用时，优先使用gpu。也可以通过以下方式设置环境变量`MGE_DEFAULT_DEVICE`。