chainer如何使用multi-cpu之类的multi-gpu

时间:2018-02-21 01:48:34

标签: chainer

在chainer.dataset.to_device中,我找到了

参数:

device(int或None) - 要将数组发送到的设备ID。如果是负值,则将数组发送到CPU。如果是正数,则使用给定ID将数组发送到GPU。如果为None,则在原始设备中保留一个数组。 x(numpy.ndarray或cupy.ndarray) - 要发送的数组。

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

1 个答案:

答案 0 :(得分:0)

Chainer不区分多个CPU设备。 对于chainer.datset.to_devicedocs),如果device参数为负,则数组将转换为CPU数组,无论值是什么。