Pytorch 框架
seed = 123
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
TensorFLow / Keras 框架
seed = 123
tf.set_random_seed(seed)
np.random.seed(seed)
seed = 123
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
seed = 123
tf.set_random_seed(seed)
np.random.seed(seed)