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

30 ビュー (過去 30 日間)
Kouta Uchimoto
Kouta Uchimoto 2019 年 10 月 9 日
回答済み: Kouta Uchimoto 2019 年 10 月 12 日
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

採用された回答

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

その他の回答 (1 件)

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

Community Treasure Hunt

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

Start Hunting!