Update

Author: svekars

conf.py

@@ -102,13 +102,13 @@ def reset_seeds(gallery_conf, fname):
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark = False
     torch._dynamo.reset()
+    torch._inductor.config.force_disable_caches = True
     torch.manual_seed(42)
     torch.set_default_device(None)
     random.seed(10)
     numpy.random.seed(10)
     torch.set_grad_enabled(True)
 
-
     gc.collect()
 
 sphinx_gallery_conf = {

intermediate_source/memory_format_tutorial.py

@@ -379,17 +379,10 @@ def attribute(m):
 import gc
 import sys
 
-torch.Tensor = old_attrs[torch.Tensor]["Tensor"]
-torch.nn.functional = old_attrs[torch.nn.functional]
-for attr_name in ["is_cuda", "has_names", "numel", "stride", "is_contiguous", "__class__"]:
-    if attr_name in old_attrs[torch]:
-        setattr(torch, attr_name, old_attrs[torch][attr_name])
+if torch.nn.functional in old_attrs:
+    for k, v in old_attrs[torch.nn.functional].items():
+        setattr(torch.nn.functional, k, v)
 
-# Force garbage collection
-del old_attrs
-gc.collect()
-
-# Reset torch modules
 torch._dynamo.reset()
 if torch.cuda.is_available():
     torch.cuda.empty_cache()