如何在PyTorch中为MPS设备设置手动随机种子？

PyTorch中针对MPS设备的随机种子设置，无需调用torch.backends.mps.manual_seed()，正确操作方式如下：

1. 全局统一设置（覆盖CPU与MPS设备） #

直接使用torch.manual_seed()即可，该方法会同步设置CPU和所有可用MPS设备的随机种子：

import torch

if torch.backends.mps.is_available():
    torch.manual_seed(0)

2. 单独为MPS设备设置种子 #

若需单独指定MPS设备的种子，可使用torch.mps.manual_seed()方法：

import torch

if torch.backends.mps.is_available():
    torch.mps.manual_seed(0)

额外注意事项 #

• 若使用的PyTorch版本较旧，可能没有torch.mps.manual_seed()，此时仅用全局的torch.manual_seed()即可，它会自动同步MPS设备的种子状态。
• 若要确保实验完全可复现，除了设备种子外，建议同时设置Python内置随机库、NumPy的种子（如果项目中用到这些库）：

import random
import numpy as np

random.seed(0)
np.random.seed(0)
torch.manual_seed(0)
if torch.backends.mps.is_available():
    torch.mps.manual_seed(0)

内容的提问来源于stack exchange，提问作者Muri7910