diff --git a/conf.py b/conf.py index b70d79b9df6..896fbc1608f 100644 --- a/conf.py +++ b/conf.py @@ -87,6 +87,7 @@ # -- Sphinx-gallery configuration -------------------------------------------- def reset_seeds(gallery_conf, fname): + torch.cuda.empty_cache() torch.manual_seed(42) torch.set_default_device(None) random.seed(10)