Multi-GPU Usage:caffe下的多GPU同时使用
最近在跑FCN的时候发现caffe.set_device()只能有一个int参数,导致只能用一个GPU跑,于是开始着手解决。
caffe声明:https://github.com/BVLC/caffe/blob/master/docs/multigpu.md
截图如下:
多GPU接口只支持C/C++,以及训练。
正在绞尽脑汁解决~
最近在跑FCN的时候发现caffe.set_device()只能有一个int参数,导致只能用一个GPU跑,于是开始着手解决。
caffe声明:https://github.com/BVLC/caffe/blob/master/docs/multigpu.md
截图如下:
多GPU接口只支持C/C++,以及训练。
正在绞尽脑汁解决~