torch.cuda.seed_all

torch.cuda.seed_all()

在所有 GPU 上將隨機數生成器的種子設定為一個隨機數。

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