M1 Mac Mini 上的 Tensorflow 2.5 比之前的 Tensorflow-for-Mac 版本慢得多

时间:2021-06-30 16:59:27

标签: python macos tensorflow mnist apple-m1

我正在尝试对我的 M1 Mac Mini 进行一些简单的基准测试。我曾经使用过 on the Apple GitHub page 中描述的 Tensorflow-for-mac 版本。最近它停止工作,我发现苹果归档了提到的存储库。他们现在指的是 this page on their developer instructions。 按照说明操作后,我尝试在此处运行这个超级简单的 MNIST 设置:

Model: "sequential"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
flatten (Flatten)            (None, 784)               0         
_________________________________________________________________
dense (Dense)                (None, 128)               100480    
_________________________________________________________________
dense_1 (Dense)              (None, 10)                1290      
=================================================================
Total params: 101,770
Trainable params: 101,770
Non-trainable params: 0
_________________________________________________________________

令我惊讶的是,与之前的 TF 版本相比,这个简单模型的训练时间长了大约 10 倍。 每个 epoch 大约需要 10 秒(以前最多需要 1 秒)。

我想我的问题是,如果这是一个已知的错误或问题。有没有办法来解决这个问题?以前的版本将不再适用于我的机器。 任何帮助表示赞赏!

1 个答案:

答案 0 :(得分:0)

这可能不是一个很好的修复,但您可以尝试为 M1 https://www.parallels.com/ 运行此 VM,然后再运行它。不知道它是否会大大提高你的性能,但我用它玩了一个没有在 M1 mac 上运行但在 intel mac 上运行的游戏