th512的博客

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

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