RuntimeError: CUDA out of memory
最新推荐文章于 2024-08-08 03:12:58 发布
在运行深度学习代码时遇到CUDA out of memory错误。经过排查,不是tensorflow和pytorch冲突,也不是设备号问题,而是batchSize过大。通过减小batchSize到50,成功解决了GPU显存不足的问题。分析原因可能是程序一次性加载数据过多,导致显存溢出。
摘要由CSDN通过智能技术生成