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

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

Answer 1

[Naoya]

頂きました gpuDevice と同程度のメモリを持つ GPU環境で実行してみましたところ、同様にMniBatchSize = 1 でメモリエラーの再現ができました。

一つの方法として、 trainingOptions で solver を "adam" から "sgdm" に変更すると、MniBatchSize = 32 程度で実行できましたので、宜しければお試しください。

Answer 2

[Kouta Uchimoto]

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