|
发表于 2023-8-8 22:27:19
|
显示全部楼层
大佬 训练时候跳出错误
Error: 2 root error(s) found.
(0) Resource exhausted: OOM when allocating tensor with shape[4,144,160,160] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[node Conv2D_17 (defined at D:\ai\DeepFaceLab_NVIDIA_RTX3000_series_build_11_20_2021\DeepFaceLab_NVIDIA_RTX3000_series\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:101) ]]
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. This isn't available when running in Eager mode.
这个是什么问题呢 新手刚使用。 |
|