我正在为 matlab 寻找基于 GPU 的深度学习库。你知道任何或任何重要的建议吗？
问问题
3780 次
2 回答
5
你能试试skaae/DeepLearnToolbox 吗？它是rasmusbergpalm/DeepLearnToolbox的一个分支，并为 GPU 添加了一些功能。它仍在开发中，所以如果它按预期工作，请回复。
于 2013-03-31T09:59:59.790 回答
4
我还没有尝试过上一个答案中提到的两种解决方案。但我可以证明matconvnet非常棒。它同时具有 CPU 和 GPU 实现。它已经过测试，可以在 linux、mac 和 windows 上的 gpu 上编译和运行。它的代码可读性很强，并且支持对网络拓扑的任意更改。
最重要的是，该代码包含大量示例，用于在已知数据集（MNIST、imagenet）上训练解决方案，并提供了一些与 caffe 的数值精度实现一致的预训练网络。
顺便说一句，我不以任何方式隶属于 matconvnet 创建者。只是一个粉丝:-)
于 2015-01-12T17:38:31.457 回答