有没有可以在GPU上运行的神经网络Matlab工具箱？

Question

我尝试使用GPUmat，但mathworks的神经网络工具箱不支持它。否则我必须自己更改nn-toolbox。但对我来说太难了。对我有什么建议吗？

Answer 1

我不知道这是否会特别加速神经网络工具箱，但Mathworks现在通过并行计算工具箱提供CUDA GPU支持：

http://www.mathworks.com/discovery/matlab-gpu.html?s_cid=HP_MI_tech_gpu

Answer 2

Matlab提供了自己的工具箱，用于在GPU上训练神经网络，请参阅here。 作为作者，我还建议使用我的工具箱ConvNet，它使用Alex Krizhevsky的库cuda-convnet2的内核。它还具有纯CPU和Matlab版本，它们的工作方式完全相同。还有另一个用于Matlab的工具箱，名为MatConvNet，但我没有检查它。