CUDA_ERROR_OUT_OF_MEMORY
4 次查看(过去 30 天)
显示 更早的评论
i am training 318 images of 1024 1024 1 size. These are the properties of my GPU.
Name: 'Quadro K6000'
Index: 1
ComputeCapability: '3.5'
SupportsDouble: 1
DriverVersion: 9
ToolkitVersion: 8
MaxThreadsPerBlock: 1024
MaxShmemPerBlock: 49152
MaxThreadBlockSize: [1024 1024 64]
MaxGridSize: [2.1475e+09 65535 65535]
SIMDWidth: 32
TotalMemory: 1.2885e+10
MultiprocessorCount: 15
ClockRateKHz: 901500
ComputeMode: 'Default'
GPUOverlapsTransfers: 1
KernelExecutionTimeout: 1
CanMapHostMemory: 1
DeviceSupported: 1
DeviceSelected: 1
I am using minibatchsize '5'.
layers = [
imageInputLayer([1024 1024 1]);
convolution2dLayer(3,16)
batchNormalizationLayer;
reluLayer();
averagePooling2dLayer(2,'Stride',2);
dropoutLayer
convolution2dLayer(3,32);
batchNormalizationLayer;
reluLayer();
averagePooling2dLayer(2,'Stride',2);
dropoutLayer
fullyConnectedLayer(2);
softmaxLayer();
classificationLayer()];
I get CUDA out of memory error.Help please.
3 个评论
Joss Knight
2018-8-8
编辑:Joss Knight
2018-8-8
All the standard networks use ImageNet data at 227x227 or 224x224. Can you upgrade MATLAB?
回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!