如何使用pytorch为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)
如何使用pytorch为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)