random.seed(args.seed)
np.random.seed(args.seed)

torch.cuda.manual_seed(args.seed)
torch.manual_seed(args.seed)
torch.backends.cudnn.deterministic = True

env = gym.make(env_name).unwrapped
env.seed(args.seed)

 

相关文章: