torch.cuda.seed

torch.cuda.seed()

設定目前 GPU 產生隨機數的種子為一個隨機數。

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

警告

如果您正在使用多 GPU 模型，此函數只會初始化一個 GPU 上的種子。要初始化所有 GPU，請使用 seed_all()。