M1 Mac Mini上的Tensorflow 2.5比以前的TensorflowforMac版本慢得多

2024-03-29 14:19:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在我的M1 Mac Mini上做一些简单的基准测试。我曾经使用Tensorflow for mac版本,如on the Apple GitHub page所述。最近它停止工作了,我发现苹果公司归档了提到的存储库。它们现在指的是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倍左右。 每个历元大约需要10秒(以前最多需要1秒)

我想我的问题是,如果这是一个已知的bug或问题。有办法解决这个问题吗?以前的版本在我的机器上不再工作。 感谢您的帮助


Tags: the版本noneforonmactensorflowpage
1条回答
网友
1楼 · 发布于 2024-03-29 14:19:36

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

相关问题 更多 >