th512的博客

原创 踩过cuda、cudnn的坑[持续更新]

加载torch模型，测试数据导致内存溢出，RuntimeError: CUDA out of memory. # 内存溢出运行一段时间，程序再崩：那估计就是代码中有加载到内存中的东西没有释放。如 运行plt模块会造成内存溢出，估计是plt模块未释放：plt.close("all")显卡问题的解决方案：# 禁用cudnntorch.backends.cudnn.benchmar...