Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add tensorflow-macos conda package #2

Merged
merged 62 commits into from
Sep 11, 2024

Conversation

eberrigan
Copy link
Collaborator

Summary

  • Currently, tensorflow-macos is pip-installed in SLEAP.
  • This approach results in the SLEAP conda package for macOS being very large.
  • We already maintain a tensorflow conda package for Windows/Linux.
  • We propose to maintain a tensorflow-macos conda package for SLEAP to use when building the SLEAP conda package or creating a SLEAP development environment on Apple silicon macs.
  • All SLEAP conda packages (for different OS') will then be similarly sized.
  • Maintaining tensorflow conda packages separately from the SLEAP conda packages adds flexibility to our builds.

Implementation

  1. Create tensorflow-macos Conda Package:

    • Using `tensorflow-macos == 2.9.2 and Python 3.10.
    • Figure out dependencies.
    • Make meta.yaml and build.sh.
  2. Test tensorflow-macos Conda Package

    • Download package.
    • Create environment with package.
    • Make sure tensorflow-macos can be imported and find local GPUs.
  3. Update SLEAP Conda Build:

    • Modify the SLEAP conda build process to use the new tensorflow-macos conda package when building on Apple silicon macs.
    • Test with CI.

eberrigan and others added 26 commits July 17, 2024 11:00
@roomrys roomrys merged commit 40bb6fd into main Sep 11, 2024
1 check passed
@roomrys roomrys deleted the elizabeth/add-tensorflow-macos-conda-package branch September 12, 2024 15:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants