chainer - chainer 如何像 multi-gpu 一样使用多 CPU

Question

在chainer.dataset.to_device，我发现

参数：

device (int or None) -- 向其发送数组的设备 ID。如果为负值，则将数组发送到 CPU。如果为正，则将具有给定 ID 的数组发送到 GPU。如果为 None，则在原始设备中留下一个数组。x (numpy.ndarray 或 cupy.ndarray) -- 要发送的数组。

chainer 使用 0,1,2... 来表示 gpus 设备 ID。但是对于 cpu，我可以使用数字 -1,-2,... 来代表我要选择的不同 cpu 设备吗？

Answer 1

Chainer 不区分多个 CPU 设备。至于chainer.datset.to_device（docsdevice ），如果参数为负，则无论值是什么，数组都将转换为 CPU 数组。