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

5 次查看（过去 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
わざわざ検証までしていただきありがとうございます ものすごく助かりました

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

产品


版本

R2019a

Community Treasure Hunt

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

Start Hunting!