Skip to content

Pin TensorFlow according to OS; Fix DeepLabCut install in Colab to deal with new requirements. #2862

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.

Sign up for GitHub

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

Jump to bottom
Merged
MMathisLab merged 3 commits into from
Feb 7, 2025
Merged

Pin TensorFlow according to OS; Fix DeepLabCut install in Colab to deal with new requirements. #2862

MMathisLab merged 3 commits into from
Feb 7, 2025

Conversation

@n-poulsen
Copy link
Contributor

@n-poulsen n-poulsen commented Feb 6, 2025

This pull request sets the TensorFlow pin to version 2.10 for windows, and version <=2.12 for other OSs. It also updates the installation method for DeepLabCut in COLAB notebooks, to match current requirements. The updated install is faster than the old install. To TensorFlow 2.12 working with the GPU in COLAB, one needs to run (in this order):

# Cell 1 - Install the desired TensorFlow version, built for CUDA 11.8 and cudnn 8
!pip install "tensorflow==2.12.1" "tensorpack>=0.11" "tf_slim>=1.1.0"

# Cell 2 - Downgrade PyTorch to a version using CUDA 11.8 and cudnn 8
#   This should also install the required CUDA libraries, for both PyTorch and TensorFlow
!pip install torch==2.3.1 torchvision --index-url https://download.pytorch.org/whl/cu118

# Cell 3
# As described in https://www.tensorflow.org/install/pip#step-by-step_instructions:
# Create symbolic links to NVIDIA shared libraries
!ln -svf /usr/local/lib/python3.11/dist-packages/nvidia/*/lib/*.so* /usr/local/lib/python3.11/dist-packages/tensorflow

Then DeepLabCut can be installed with the desired version (e.g. pip install deeplabcut or pip install "git+https://github.com/DeepLabCut/DeepLabCut.git)

Closes #2839

maximpavliv
maximpavliv approved these changes Feb 7, 2025
View reviewed changes
Copy link
Contributor

@maximpavliv maximpavliv left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Perfect, thanks a lot @n-poulsen ! Let's merge !

@MMathisLab MMathisLab merged commit c483526 into main Feb 7, 2025
5 checks passed
@MMathisLab MMathisLab deleted the niels/tf_pin_linux branch February 7, 2025 09:28
@maximpavliv maximpavliv mentioned this pull request Sep 10, 2025
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@MMathisLab MMathisLab MMathisLab approved these changes

+1 more reviewer

@maximpavliv maximpavliv maximpavliv approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

COLAB Installation Tensorflow 2

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

COLAB || AttributeError: BatchNormalization is not available with Keras 3.

4 participants

@n-poulsen @MMathisLab @maximpavliv