清空无引用的Gpu显存占用 

torch.cuda.empty_cache()
# 服务器的内存(主存)最大可达几十T,因此服务器上可以做到直接将所有训练数据加载到内存之后训练
posted @ 2020-11-27 14:46  Parallax  阅读(388)  评论(0)  编辑  收藏  举报