Apple M1 和其他设备上的 TensorFlow 的通用要求.txt

CGF*_*FoX 6 python macos requirements.txt tensorflow apple-m1

我有一台配备 Apple M1 芯片组的新 MacBook。要安装tensorflow,我按照此处的说明进行操作,即安装tensorflow-metalandtensorflow-macos而不是普通的tensorflow软件包。

虽然这工作正常,但这意味着我无法运行典型的,pip install -r requirements.txt只要我们tensorflowrequirements.txt. 如果我们改为包含tensorflow-macos,则会给非 M1 甚至非 macOS 用户带来问题。

我们的库必须在所有平台上运行。是否有一个通用安装命令可以根据计算机是否是 M1 Mac 来安装正确的 TensorFlow 版本?这样我们就可以为requirements.txt每个人使用一个单一的?

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

小智 7

根据这篇文章Is有一种方法可以为我的基于平台的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'
Run Code Online (Sandbox Code Playgroud)