Pergunta

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

Foi útil?

Solução

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.

Outras dicas

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 :-)

Licenciado em: CC-BY-SA com atribuição
Não afiliado a StackOverflow
scroll top