ディープラーニング入門 GPUメモリが足りない

12 visualizzazioni (ultimi 30 giorni)
Kouta Uchimoto
Kouta Uchimoto il 9 Ott 2019
Risposto: Kouta Uchimoto il 12 Ott 2019
matlabのディープラーニング入門の最終プロジェクト線虫の生死確認のサンプルプログラムを実行したのですが、
エラー:GPUのメモリが足りません 関数trainingOptionsを使用してMiniBatchSizeを減らしてみてください。
原因:デバイスのメモリが足りません。GPUで利用可能なメモリを表示するにはgpuDevice()を使用します。問題が解決しない場合はgpuDevice(1)を呼び出してGPUをリセットしてください。
とでますMiniBatchSizeを64や1にしてもエラーが解決できず困っています。
使っているGPUはGUNFORRCE 960です
CUDADevice のプロパティ:
Name: 'GeForce GTX 960'
Index: 1
ComputeCapability: '5.2'
SupportsDouble: 1
DriverVersion: 10.1000
ToolkitVersion: 10.1000
MaxThreadsPerBlock: 1024
MaxShmemPerBlock: 49152
MaxThreadBlockSize: [1024 1024 64]
MaxGridSize: [2.1475e+09 65535 65535]
SIMDWidth: 32
TotalMemory: 2.1475e+09
AvailableMemory: 1.5060e+09
MultiprocessorCount: 8
ClockRateKHz: 1240500
ComputeMode: 'Default'
GPUOverlapsTransfers: 1
KernelExecutionTimeout: 1
CanMapHostMemory: 1
DeviceSupported: 1
DeviceSelected: 1

Risposta accettata

Naoya
Naoya il 10 Ott 2019
頂きました gpuDevice と同程度のメモリを持つ GPU環境で実行してみましたところ、同様にMniBatchSize = 1 でメモリエラーの再現ができました。
一つの方法として、 trainingOptions で solver を "adam" から "sgdm" に変更すると、MniBatchSize = 32 程度で実行できましたので、宜しければお試しください。

Più risposte (1)

Kouta Uchimoto
Kouta Uchimoto il 12 Ott 2019
わざわざ検証までしていただきありがとうございます ものすごく助かりました

Categorie

Scopri di più su 並列およびクラウドでの深層学習 in Help Center e File Exchange

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!