English

Theano-based Large-Scale Visual Recognition with Multiple GPUs

Machine Learning 2015-04-08 v4

Abstract

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.

Cite

@article{arxiv.1412.2302,
  title  = {Theano-based Large-Scale Visual Recognition with Multiple GPUs},
  author = {Weiguang Ding and Ruoyan Wang and Fei Mao and Graham Taylor},
  journal= {arXiv preprint arXiv:1412.2302},
  year   = {2015}
}

Comments

ICLR 2015 workshop camera-ready version

R2 v1 2026-06-22T07:22:38.562Z