Question

I am looking for a GPU based Deep Learning library for matlab. Do you know any or any suggestions that matter?

Was it helpful?

Solution

Can you try skaae/DeepLearnToolbox. It is a fork from rasmusbergpalm/DeepLearnToolbox and has some added functionality for GPU. It is still under development so reply if it works as expected.

OTHER TIPS

I haven't tried the two solutions that are mentioned in a previous answer. But I can attest that matconvnet is superb. It has both CPU and GPU implementations. It has been tested to compile and run on a gpu on both linux, mac and windows. Its code is very readable, and it supports arbitrary changes to the network's topology.

On top of it all, the code includes ample examples for training solution on known datasets (MNIST, imagenet), and provides some pretrained networks that are consistent with caffe's implementation to numerical precision.

I'm not affiliated with matconvnet creators in any way, by the way. Just a fan :-)

Licensed under: CC-BY-SA with attribution
Not affiliated with StackOverflow
scroll top