[TensorFlow] 错误之tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 150994944 exceeds

2023-05-16

在tensorflow2.0使用cpu的时候出现警告信息

tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 150994944 exceeds 

此时会发现gpu能被占用少量显存但无法使用,导致使用cpu运行(速度慢)

建议降低batch_size等,或释放些内存增加可用内存。

参考:https://github.com/tensorflow/tensorflow/issues/18736

未完待续……

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

[TensorFlow] 错误之tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 150994944 exceeds 的相关文章

随机推荐