火花灯中mps器件的manual_seed

Question

我如何建立一个手动种子的mps设备使用火把？对于cuda设备，代码应该是这样工作的：

if torch.cuda.is_available():
  torch.cuda.manual_seed(0)

我用的是苹果m1芯片。以下语句返回True：

torch.backends.mps.is_available()

但以下声明是不可能的：

torch.backends.mps.manual_seed(0)

Answer 1

根据文档

您可以使用torch.manual_seed()为所有设备( CPU和CPU)设置RNG种子：

import torch
torch.manual_seed(0)