转载请注明出处:小锋学长生活大爆炸[xfxuezhagn.cn] 如果本文帮助到了你,欢迎[点赞、收藏、关注]哦~
import torch
import random
import numpy as np
def set_random_seed(seed):
# 设置Python的随机种子
random.seed(seed)
# 设置NumPy的随机种子
np.random.seed(seed)
# 设置CPU的随机种子
torch.manual_seed(seed)
# 设置当前GPU设备的随机种子
torch.cuda.manual_seed(seed)
# 设置所有GPU设备的随机种子(如果使用多GPU)
torch.cuda.manual_seed_all(seed)
# 禁用CuDNN中可能导致非确定性行为的选项
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
# 设置随机种子
seed = 42
set_random_seed(seed)
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。