torch.cuda.seed

torch.cuda.seed()

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

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

警告

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