torch.mps.empty_cache()

torch.mps.empty_cache()[源代码]

释放缓存分配器中所有未使用的缓存内存，使这些内存可供其他 GPU 应用程序使用。