Skip to content

torch.cuda.empty_cache() 不能释放显存 #130

Open
@smoftsky

Description

@smoftsky

用了torch.cuda.empty_cache()后,在控制台nvidia-smi查看显存没有释放,在Ipython控制台继续添加网络后显存继续上升。是我对相关的内存机制理解有问题,还是是因为我用的是spyder而和spyder运行特性有关呢?求助大佬。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions