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,Pytorch)