0

I have a package that depends on tensorflow. However, to use tensorflow on an Apple silicon computer, users need to install tensorflow-macos. I'm wondering if there is a neat way to specify this difference in setup.py to make it easy for the end user?

For example, the best I have thought of is to use extra_requires, so that a user could use pip install package[intel] to install tensorflow, and pip install package[macos] to install tensorflow-macos. However, I'm not a huge fan of this solution. My ideal would be that there is a default installation (e.g. pip install package) that can recognize the environment it is being installed on and select the appropriate tensorflow package. Is this possible to specify in setup.py?

sjeknic
  • 15
  • 5
  • https://stackoverflow.com/questions/65970469/what-does-platform-system-and-platform-architecture-return-on-apple-m1-silic – Martheen Sep 24 '22 at 17:13

0 Answers0