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: CUD