pytorch - 💡(How to fix) Fix `set_multithreading_enabled` still spawns autograd threads

import os, torch

def autograd_threads():
    """Return list of native thread names containing 'autograd' (Linux)."""
    names = []
    for tid in os.listdir("/proc/self/task"):
        try:
            name = open(f"/proc/self/task/{tid}/comm").read().strip()
            if "autograd" in name:
                names.append(name)
        except OSError:
            pass
    return names

torch.autograd.grad_mode.set_multithreading_enabled(False)
print(f"autograd threads before backward: {autograd_threads()}")
x = torch.ones(100, requires_grad=True)
x.mean().backward()
threads = autograd_threads()
print(f"autograd threads after backward:  {threads}")

---

autograd threads before backward: []
autograd threads after backward:  ['pt_autograd_0', 'pt_autograd_1', 'pt_autograd_2', 'pt_autograd_3', 'pt_autograd_4', 'pt_autograd_5', 'pt_autograd_6', 'pt_autograd_7']