Theano-based Large-Scale Visual Recognition with Multiple GPUs

Weiguang Ding, Ruoyan Wang, Fei Mao, Graham Taylor
School of Engineering, University of Guelph
arXiv:1412.2302 [cs.LG], (7 Dec 2014)



In this report, we describe a Theano-based AlexNet (Krizhevsky et al., 2012) implementation and its naive data parallelism on multiple GPUs. Our performance on 2 GPUs is comparable with the state-of-art Caffe library (Jia et al., 2014) run on 1 GPU. To the best of our knowledge, this is the first open-source Python-based AlexNet implementation to-date.
No votes yet.
Please wait...

* * *

* * *

HGPU group © 2010-2021 hgpu.org

All rights belong to the respective authors

Contact us: