سؤال

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

هل كانت مفيدة؟

المحلول

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.

نصائح أخرى

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

مرخصة بموجب: CC-BY-SA مع الإسناد
لا تنتمي إلى StackOverflow
scroll top