py-R-FCN icon indicating copy to clipboard operation
py-R-FCN copied to clipboard

Does this project support trained by multiple GPUs?

Open chungjin opened this issue 9 years ago • 3 comments

I wander whether your project can be trained by multiple GPUs? I have tried to use the command like --gpu 0,1,2 but it seems it does not support this way. But in Dai's paper, they use 8 GPUs to train the model.

chungjin avatar Sep 18 '16 03:09 chungjin

I don't think this is possible, since it used python layer, and caffe does not support multi-gpu if python layer is used

zhangjiangqige avatar Sep 27 '16 07:09 zhangjiangqige

I think so, so they must have used the C++ and Cuda to rewrite the python layer

chungjin avatar Sep 28 '16 02:09 chungjin

https://github.com/bharatsingh430/py-R-FCN-multiGPU

You can use this branch

bharatsingh430 avatar Jan 24 '17 20:01 bharatsingh430