pytorch在M1上的段错误:torch.from_numpy(X).float()



我用的是M1。

我正在尝试使用pytorch的转换网络。

我有一个numpy数组,我想把它变成一个火炬张量。

当我呼叫

torch.from_numpy(X)

pytorch会抛出一个错误,认为它得到的是双精度浮点数。

当我呼叫

torch.from_numpy(X).float()在一个朋友的电脑上,一切都很好。

但是当我在我的计算机上调用这个命令时,我得到一个段错误。

有没有人看到这个/知道什么可能发生/知道如何修复?

你的pytorch愿景是什么?我在Macbook Pro M1上遇到了同样的问题,我的pytorch版本最初是1.12.0。我把它降级到1.10.0版本,问题就解决了。我怀疑这与新版本的火炬与M1的兼容性有关。

实际上我先用pip3 uninstall torch卸载火炬,然后用pip3 install torch==1.10.0重新安装

但是如果你正在使用火炬视觉或其他附属软件包,你可能也需要降级它们。