我有一台带有苹果M1芯片组的新MacBook。要安装tensorflow,我按照这里的说明进行操作,即安装tensorflow-metal
和tensorflow-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'