pytorch CUDA out of memor 解决方法: 用完把cuda变量 del 显存能增大一倍 import time net = BiSeNet() net.cuda() net.eval() in_ten = torch.randn(8, 3, 640, 640).cuda() for i in range(8): start=time.time() out, out16, out32 = net(in_ten) print(time.time()-start,out.shape,out16.shape,out32.shape) del out del out16 del out32 来源: oschina 链接: https://my.oschina.net/u/4271034/blog/4277811