torch.manual_seed() 和 torch.cuda.manual_seed() 功能及实例

最新推荐文章于 2024-02-21 16:40:08 发布

原创最新推荐文章于 2024-02-21 16:40:08 发布 · 1w 阅读

20 ·

CC 4.0 BY-SA版权

PyTorch 基础专栏收录该内容

10 篇文章

订阅专栏

本文介绍如何在PyTorch中设置随机数种子，确保每次运行生成相同的随机数，适用于CPU和GPU环境。通过设置torch.manual_seed和torch.cuda.manual_seed，实现实验结果的可复现性。

部署运行你感兴趣的模型镜像

功能

设置固定生成随机数的种子，使得每次运行该 .py 文件时生成的随机数相同

设置方法

CPU：torch.manual_seed(整数)
- 官方文档：https://pytorch.org/docs/stable/torch.html?highlight=manual_seed#torch.manual_seed
GPU：torch.cuda.manual_seed(整数)
- 官方文档：https://pytorch.org/docs/stable/cuda.html?highlight=manual_seed#torch.cuda.manual_seed

实例

# 需要注意不要在终端中单行敲入运行如下代码，要将如下代码先拷贝到 *.py 文件中，再在终端命令中通过 python *.py 运行

import torch

if torch.cuda.is_available():
    print("gpu cuda is available!")
    torch.cuda.manual_seed(1000)
else:
    print("cuda is not available! cpu is available!")
    torch.manual_seed(1000)

print(torch.rand(1, 2))

您可能感兴趣的与本文相关的镜像