在PyTorch中设置随机数生成器的种子值

2023-12-26 15:45:29

在PyTorch中设置随机数生成器的种子值的方法

CPU:可以使用torch.manual_seed()函数

GPU:需要额外设置torch.cuda.manual_seed()来设置GPU上的随机数生成器种子值。

import torch

# 设置随机种子
torch.manual_seed(0)

if torch.cuda.is_available():
    torch.cuda.manual_seed(0)

这样设置种子值后,无论是CPU还是GPU上的随机数生成器都将产生相同的随机数序列。

文章来源:https://blog.csdn.net/m0_37738114/article/details/135222271
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。