tensorflow输出固定结果

1.CPU

在代码开始部分

import os
import tensorflow as tf
import numpy as np
import random
seed=2
random.seed(seed)# 为python设置随机种子
os.environ['PYTHONHASHSEED'] = str(seed)
np.random.seed(seed)# 为numpy设置随机种子
# tf.set_random_seed(seed) # for tf1
tf.compat.v1.set_random_seed(seed)

2. GPU

#please `pip install tensorflow-determinism` first
import os
import tensorflow as tf
import numpy as np
import random
seed=2
random.seed(seed)# 为python设置随机种子
os.environ['PYTHONHASHSEED'] = str(seed)
np.random.seed(seed)# 为numpy设置随机种子
tf.compat.v1.set_random_seed(seed)# tf cpu fix seed
os.environ['TF_DETERMINISTIC_OPS'] = '1'  # tf gpu fix seed

参考：

TensorFlow2.x GPU版本固定随机种子seed，使得实验结果可复现_白马金羁侠少年的博客-CSDN博客