是否有基于GPU的Matlab工具箱用于深度学习?

时间:2013-03-31 09:15:13

标签: matlab machine-learning deep-learning

我正在为matlab寻找基于GPU的深度学习库。你知道任何重要的建议吗?

2 个答案:

答案 0 :(得分:5)

你能试试skaae/DeepLearnToolbox吗?它是rasmusbergpalm/DeepLearnToolbox的一个分支,并为GPU提供了一些附加功能。它仍在开发中,如果按预期工作则回复。

答案 1 :(得分:4)

我还没有尝试过上一个答案中提到的两个解决方案。但我可以证明matconvnet非常棒。它同时具有CPU和GPU实现。它已经过测试,可以在linux,mac和windows上的gpu上编译和运行。它的代码非常易读,它支持对网络拓扑的任意更改。

最重要的是,代码包含了大量已知数据集(MNIST,imagenet)训练解决方案的示例,并提供了一些与数值精度的caffe实现一致的预训练网络。

顺便说一句,我没有以任何方式与matconvnet创作者联系。只是一个粉丝: - )