2013-03-31 49 views

回答

5

你可以試試skaae/DeepLearnToolbox。這是一個來自rasmusbergpalm/DeepLearnToolbox的分支,併爲GPU增加了一些功能。它仍在開發中,因此如果它按預期工作就會回覆。

+0

也許這是我的不好,但我看不到和GPU利用率至少在自述文件解釋 – erogol

+0

skaae和rasmusbergpalm具有相同的自述文件,它並沒有說明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實現。它已經過測試,可以在linux,mac和windows上的gpu上編譯和運行。它的代碼非常易讀,並且支持對網絡拓撲的任意更改。除此之外,代碼還包含了大量用於對已知數據集(MNIST,imagenet)進行培訓解決方案的示例,並提供了一些與caffe的實現一致的數學精確度的預訓練網絡。順便說一下,我不以任何方式隸屬於matconvnet的創作者。只是一個粉絲:-)

相關問題