Skip to content

Instantly share code, notes, and snippets.

@jangsoopark

jangsoopark/pytorch_random_seed.py

Created July 25, 2024 15:21
Show Gist options
  • Save jangsoopark/106fcaa332726c60ac8fb021ec42d2d3 to your computer and use it in GitHub Desktop.
Save jangsoopark/106fcaa332726c60ac8fb021ec42d2d3 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pytorch_random_seed.py
import numpy as np
import torch
import random
def set_random_seed(random_seed):
torch.manual_seed(random_seed)
torch.cuda.manual_seed(random_seed)
torch.cuda.manual_seed_all(random_seed) # if use multi-GPU
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
np.random.seed(random_seed)
random.seed(random_seed)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment