2023-06-09 14:36:39.705522: W tensorflow/core/common_runtime/bfc_allocator.cc:479] Allocator (GPU_0_bfc) ran out of memory trying to allocate 768.00MiB (rounded to 805306368)requested by op AddV2 If the cause is memory fragmentation maybe the environment variable 'TF_GPU_ALLOCATOR=cuda_malloc_async' will improve the situation.
时间: 2024-04-04 18:31:10 浏览: 199
front-end-interview-prepare:记录一些前端面试知识点
这个警告是 TensorFlow 运行时发出的,表示在分配显存时出现了内存不足的情况。可能是因为模型太大或者数据量太大,导致显存不够。警告中提到可以尝试设置环境变量 `TF_GPU_ALLOCATOR=cuda_malloc_async` 来解决内存碎片化的问题,进而提高显存使用效率。如果这个警告频繁出现,可以考虑减小模型规模或者使用更大的显存。
阅读全文