本文主要是介绍pytorch 训练正常,测试显存爆掉问题解决，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
pytorch 训练正常,测试显存爆掉问题解决
- Intro
- pytorch version
- cudnn version
- pytorch训练正常但是在预测的时候显存爆掉,错误信息如下
- 暂时解决方法:
- [参考链接](https://discuss.pytorch.org/t/strange-gpu-memory-behavior-strange-memory-consumption-and-out-of-memory-error/2700)
Intro
pytorch gpu out of memory when test the model
pytorch version
0.4.0
cudnn version
7.0.1
pytorch训练正常但是在预测的时候显存爆掉,错误信息如下
RuntimeError: CUDNN_STATUS_INTERNAL_ERROR
暂时解决方法:
加入:
torch.backends.cudnn.enabled = False参考链接
这篇关于pytorch 训练正常,测试显存爆掉问题解决的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
