Skip to content

Instantly share code, notes, and snippets.

@Creep-ai
Creep-ai / pytorch_seed.py
Last active July 13, 2025 15:44
set seed for pytorch everything
def seed_all(seed: int = 42) -> None:
random.seed(seed)
os.environ['PYTHONHASHSEED'] = str(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'