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

推荐答案

根据前docs名:

您可以使用torch.manuseed()为所有设备(CPU和CUDA)设定RNG种子:

import torch
torch.manual_seed(0)

Python-3.x相关问答推荐

如何通过Pandas为不同的列集垂直设置列数据?

如何检索与美汤相似的标签中的文本?

将数据帧扩展为矩阵索引

Django将任何查询显示为html表格

如何从枚举中获取某个值?

PySpark每毫秒使用先前的值填充数据

如何获取实例化 `types.GenericAlias` 的下标类?

生成具有偶数个 0 和 1 的给定长度的所有二进制数

在新数据帧上自动提取两个字符串 python 之间的相等性

它们是否同样存储在python3的内存中?

为列表列表中的每个列表插入 str 到 index[0] 中. Python

如何在 Spyder 控制台中使用变量执行 Python 3.3 脚本?

Python图例属性错误

if 语句中冒号的语法错误

如何将 SimpleGUI 与 Python 2.7 和 3.0 shell 集成

如何在 Python3 中添加带有标志的命令行参数?

创建集合的 Python 性能比较 - set() 与 {} 文字

从 IPython 重新加载 Python 扩展模块

向 Python 函数添加属性的最佳方法

在 macbook pro M1 上安装 Tensorflow 时出现zsh:非法硬件指令 python