背景
在使用FakeApp训练的时候出错,表现为点击“train”按钮后一段时间就报错,且没有训练预览窗口弹出。
log文件里显示
undefinedUsing GPU0 for processing
Focus the training preview window and press ‘q’ to stop training and save the model.ResourceExhaustedError: OOM when allocating tensor with shape[3,3,512,1024] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[Node: training_1/Adam/mul_41 = Mul[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Adam/beta_1/read, training_1/Adam/Variable_8/read)]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
配置
联系笔记本,双显卡,独显是N卡GTX950M。
(比较渣)
无效的解决方法
尝试禁用集显,但是没用
解决方法
首先要确保自己的安装过程、路径、环境