在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
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。