torch.cuda.manual_seed#
- torch.cuda.manual_seed(seed)[source]#
设置当前 GPU 随机数生成的种子。
如果 CUDA 不可用，调用此函数是安全的；在这种情况下，它将被静默忽略。
- 参数
seed (int) – 期望的种子。
警告
如果您正在处理多 GPU 模型，此函数不足以实现确定性。要为所有 GPU 设置种子，请使用
manual_seed_all()。
设置当前 GPU 随机数生成的种子。
如果 CUDA 不可用，调用此函数是安全的；在这种情况下，它将被静默忽略。
seed (int) – 期望的种子。
警告
如果您正在处理多 GPU 模型，此函数不足以实现确定性。要为所有 GPU 设置种子，请使用 manual_seed_all()。