开发者社区  >  开发问答  >  卷积升级网络图像去噪遇到了错误使用 gpuArray/max是怎么回事?

卷积升级网络图像去噪遇到了错误使用 gpuArray/max是怎么回事?

各位大佬,请问我再运行DnCNN这个代码TrainingCdoe里

DnCNN_TrainingCodes_v1.0

当batchsize为64的时候不会报错,当batchsize为128时候会报错,错误是

错误使用 gpuArray/max

Out of memory on device. To view more detail about available memory on the GPU, use 'gpuDevice()'. If the problem persists, reset the GPU by

calling 'gpuDevice(1)'.21

当batchsize在64和128之间会报错为:

错误使用 vl_nnbnorm

Out of memory on device. To view more detail about available memory on the GPU, use 'gpuDevice()'. If the problem persists, reset the GPU by

calling 'gpuDevice(1)'.

4

请问我该怎么办?


相关回答 

VueGirl 10天前 207声望

gpu显存不够了吧,

Out of memory on device. 

To view more detail about available memory on the GPU, use 'gpuDevice()'

输入 gpuDevice() 可以查看你的显存大小。

(0) 回复

lzt12138 回复 VueGirl 10天前

Name: 'GeForce MX250' Index: 1 ComputeCapability: '6.1' SupportsDouble: 1 DriverVersion: 10.2000 ToolkitVersion: 7.5000 MaxThreadsPerBlock: 1024 MaxShmemPerBlock: 49152 MaxThreadBlockSize: [1024 1024 64] MaxGridSize: [2.1475e+09 65535 65535] SIMDWidth: 32 TotalMemory: 2.1475e+09 AvailableMemory: 1.6810e+09 MultiprocessorCount: 3 ClockRateKHz: 1582000 ComputeMode: 'Default' GPUOverlapsTransfers: 1 KernelExecutionTimeout: 1 CanMapHostMemory: 1 DeviceSupported: 1 DeviceSelected: 1 我在64到128之间也试过但错误变成:错误使用 vl_nnbnorm Out of memory on device. To view more detail about available memory on the GPU, use 'gpuDevice()'. If the problem persists, reset the GPU by calling 'gpuDevice(1)'.

(0) 回复

相关问答

等待回答