我正在为matlab寻找基于GPU的深度学习库。你知道任何重要的建议吗?
答案 0 :(得分:5)
你能试试skaae/DeepLearnToolbox吗?它是rasmusbergpalm/DeepLearnToolbox的一个分支,并为GPU提供了一些附加功能。它仍在开发中,如果按预期工作则回复。
答案 1 :(得分:4)
我还没有尝试过上一个答案中提到的两个解决方案。但我可以证明matconvnet非常棒。它同时具有CPU和GPU实现。它已经过测试,可以在linux,mac和windows上的gpu上编译和运行。它的代码非常易读,它支持对网络拓扑的任意更改。
最重要的是,代码包含了大量已知数据集(MNIST,imagenet)训练解决方案的示例,并提供了一些与数值精度的caffe实现一致的预训练网络。
顺便说一句,我没有以任何方式与matconvnet创作者联系。只是一个粉丝: - )