作者:贺bujak_491 | 来源:互联网 | 2023-08-05 03:07
最近使用GPU训练一个用PyTorch写的神经网络模型,网络比较深,有100层左右,训练部分的时候没什么问题,然而开始测试/验证环节的时候,虽然我用了torch.cuda.empty_cache()
,但是竟然直接报错:
https://www.jianshu.com/p/4a2be315b32a
https://stackoverflow.com/questions/54374935/how-to-fix-this-strange-error-runtimeerror-cuda-error-out-of-memory
预测时:
with torch.no_grad():...net=Net()pred_for_validation=net(input)...