Keras/TensorFlow报错CUDA_ERROR_OUT_OF_MEMORY:

最新推荐文章于 2024-08-08 03:12:58 发布

Kyrielong

最新推荐文章于 2024-08-08 03:12:58 发布

阅读量6.2k

点赞数 4

本文链接：https://blog.csdn.net/Kyrielong/article/details/88288440

版权

在使用Keras的model.predict时遇到CUDA错误：CUDA_ERROR_OUT_OF_MEMORY。即使GPU有8GB显存，依然报内存不足。错误日志显示多次尝试分配内存失败。解决方法是通过代码限制GPU内存占用率。

摘要由CSDN通过智能技术生成

用keras的model.predict时报错如下：

failed to allocate 4.00G (4294967296 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
2019-03-07 10:24:00.610883: E tensorflow/stream_executor/cuda/cuda_driver.cc:806] failed to allocate 3.60G (3865470464 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
2019-03-07 10:24:00.613876: E tensorflow/stream_executor/cuda/cuda_driver.cc:806] failed to allocate 3.24G (3478923264 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
2019-03-07 10:24:00.617918: E tensorflow/stream_executor/cuda/cuda_driver.cc:806] failed to allocate 2.92G (3131030784 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
2019-03-07 10:24:00.621197: E tensorflow/stream_executor/cuda/cuda_driver.cc:806] failed to allocate 2.62G (2817927680 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
2019-03-07 10:24:00.624652: E tensorflow/stream_executor/cuda/cuda_driver.cc:806] failed to allocate 2.36G (2536