torch.cuda.manual_seed

torch.cuda.manual_seed(seed)

為當前 GPU 設定用於生成隨機數的種子。

如果 CUDA 不可用，呼叫此函式是安全的；在這種情況下，它會被靜默忽略。

引數

seed (int) – 期望的種子。

警告

如果您正在使用多 GPU 模型，此函式不足以保證確定性。要為所有 GPU 設定種子，請使用 manual_seed_all()。