torch.cuda.seed

torch.cuda.seed()

將當前 GPU 隨機數生成的種子設定為一個隨機數。

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

警告

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