Apple M1和其他设备上TensorFlow的Generic requirements.txt



我有一台带有苹果M1芯片组的新MacBook。要安装tensorflow,我按照这里的说明进行操作,即安装tensorflow-metaltensorflow-macos,而不是普通的tensorflow包。

虽然这很好,但这意味着只要requirements.txt中有tensorflow,我就不能运行典型的pip install -r requirements.txt。如果我们包含tensorflow-macos,这将给非M1甚至非macOS用户带来问题。

我们的图书馆必须在所有平台上工作。是否有一个通用的安装命令,根据计算机是否为M1 Mac安装正确的TensorFlow版本?这样我们就可以为每个人使用一个requirements.txt了吗?

或者,如果这不可能,我们可以传递一些标志/选项,例如pip install -r requirements.txt --m1来安装一些变体吗?这里最简单、最优雅的解决方案是什么?

根据这篇文章,有没有一种方法可以为我的Python应用程序提供一个基于平台的条件requirements.txt文件?

您可以在requirements.txt中使用条件,因此

tensorflow==2.8.0; sys_platform != 'darwin' or platform_machine != 'arm64'
tensorflow-macos==2.8.0; sys_platform == 'darwin' and platform_machine == 'arm64'

最新更新