畳み込みニューラル ネットワークを実装するために theano を使用しています。私の CPU RAM は 32G、GPU RAM は 2G ですが、データも非常に大きく、ほぼ 5G のトレーニング データです。
プログラムの実行中は、コンピューターがフリーズしているように見え、各操作が非常に遅く、応答しません。また、CPU モードは GPU モードよりも少なくとも 2 倍速いようです。
GPU 畳み込みニューラル ネットワークを高速化する方法はありますか?
質問する
586 次
2 に答える
0
GPU アクセラレーションを使用するには、最初に CUDA をインストールする必要があります。Theano 構成 (Theano フラグ/TheanoRC) のレベルでは、GPU を使用してモデルを高速化する方法はいくつかあります。
- GPUの用途を指定「device=gpu」
- Cuda メモリ割り当て (CnMem) を有効にする "cnmem = 0.75"
- CUDNN 最適化を有効にする "optimizer = cudnn"
Theano 構成の詳細については、こちらを参照してください。
于 2016-05-09T03:58:42.723 に答える
0
必ず cudnn で Theano 0.7 を使用してください。これにより、畳み込みが大幅に高速化されます。
http://deeplearning.net/software/theano/library/sandbox/cuda/dnn.html
于 2015-04-09T04:36:45.057 に答える