Skip to content

Instantly share code, notes, and snippets.

@rish-16

rish-16/tf_seed.py

Created July 5, 2021 05:30
Show Gist options
  • Star 3 You must be signed in to star a gist
  • Fork 2 You must be signed in to fork a gist
  • Save rish-16/d3566c1940f62df31919763529a01ba4 to your computer and use it in GitHub Desktop.
Save rish-16/d3566c1940f62df31919763529a01ba4 to your computer and use it in GitHub Desktop.
Download ZIP
TensorFlow Experiment Seeding for GPUs
Raw
tf_seed.py
import os
import numpy as np
import random
import tensorflow as tf
from tfdeterminism import patch
def seed(s=42):
random.seed(s)
np.random.seed(s)
tf.random.set_seed(s)
tf.experimental.numpy.random.seed(s)
tf.set_random_seed(s)
os.environ['TF_CUDNN_DETERMINISTIC'] = '1'
os.environ['TF_DETERMINISTIC_OPS'] = '1'
patch()
# https://github.com/NVIDIA/framework-determinism
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment