2013-03-31 5 views
1

Я ищу библиотеку Deep Learning для GPU для Matlab. Знаете ли вы какие-либо предложения, которые имеют значение?Есть ли какой-либо GPU на базе Matlab toolbox для глубокого обучения?

+0

Проверьте [эту ссылку] (http://www.mathworks.com/help/nnet/gs/parallel-computing.html). В нем объясняется, как использовать GPU в обучении нейронных сетей. –

ответ

5

Вы можете попробовать skaae/DeepLearnToolbox. Это вилка от rasmusbergpalm/DeepLearnToolbox и имеет некоторые дополнительные функции для графического процессора. Он все еще находится в разработке, поэтому ответьте, если он работает так, как ожидалось.

+0

Возможно, это мое плохое, но я не вижу и использования графического процессора, по крайней мере, в пояснениях Readme. – erogol

+0

В skaae и rasmusbergpalm есть одно и то же Readme, и он не говорит об использовании GPU. Файлы для GPU находятся в папке NN https://github.com/skaae/DeepLearnToolbox/tree/master/NN и папке SAE https://github.com/skaae/DeepLearnToolbox/tree/master/SAE – user568109

4

Я не пробовал два решения, упомянутые в предыдущем ответе. Но я могу подтвердить, что matconvnet превосходный. Он имеет как CPU, так и GPU. Он был протестирован для компиляции и запуска на gpu на Linux, Mac и Windows. Его код очень читабельен и поддерживает произвольные изменения топологии сети.

В дополнение ко всему, этот код включает в себя обширные примеры для обучения решениям по известным наборам данных (MNIST, imagenet) и предоставляет некоторые предварительно подготовленные сети, которые согласуются с реализацией caffe с численной точностью.

Я, кстати, не связан с создателями matconvnet. Просто поклонник :-)