探讨PyTorch在训练过程中显存使用特性，即显存占用只会增加不会减少的问题。介绍通过设置torch.backends.cudnn.enabled和benchmark参数来动态调整显存使用的技巧，但需注意这将牺牲部分训练速度。
pytorch在训练时，显存是只会增加不会减少的。如果要动态减少：
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
不过这样做的后果就是训练会很慢，快赶上CPU的速度了
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
