Multi-GPU Usage:caffe下的多GPU同时使用

最近在跑FCN的时候发现caffe.set_device()只能有一个int参数,导致只能用一个GPU跑,于是开始着手解决。

caffe声明:https://github.com/BVLC/caffe/blob/master/docs/multigpu.md

截图如下:

Multi-GPU Usage:caffe下的多GPU同时使用

多GPU接口只支持C/C++,以及训练。

正在绞尽脑汁解决~