基于 Theano 的多 GPU 大规模视觉识别
机器学习
2015-04-08 v4
摘要
在本报告中，我们描述了一种基于 Theano 的 AlexNet (Krizhevsky et al., 2012) 实现及其在多 GPU 上的朴素数据并行。我们在 2 个 GPU 上的性能可与在 1 个 GPU 上运行的当前最优 Caffe 库 (Jia et al., 2014) 相媲美。据我们所知，这是迄今为止首个开源的基于 Python 的 AlexNet 实现。
引用
@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}
}
备注
ICLR 2015 workshop camera-ready version