在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(使用前将#替换为@)