pytorch - PyTorch 在 M1 MacBook 上本地运行，但有些东西不能正常工作

Question

我刚刚拥有一台 M1 MacBook Pro，并且一直在努力设置我的开发环境。我已按照说明使用 mini-forge 安装 PyTorch，以便它可以本地运行。

为了测试这一点，我从这里运行了 MNIST 示例代码。代码运行良好，活动监视器向我显示它正在本地运行。但是，似乎仍然有问题：当在我的另一台机器上执行完全相同的代码时，通常在一个 epoch 之后才能达到 99% 的准确率，并且测试损失非常低。在 M1 Macbook 上，尽管训练损失下降，但测试损失从未低于 2，并且在 14 个 epoch 后准确率从未超过 10%。

运行其他代码时也会出现类似的问题，例如，这会在每一步都会产生 nan 的丢失。

我无法在其他任何地方找到提到的这个问题：PyTorch 在 M1 上本地运行，但它似乎无法正常工作。

对于可能导致此问题或如何解决此问题的任何建议，我将不胜感激。

谢谢！

标签： pytorchapple-m1