Skip to content

Update TensorFlow backend for Python 3.12 compatibility (Keras 2 via tf-keras) - LTS DLC 2 #3091

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

Merged
MMathisLab merged 18 commits into from
Nov 4, 2025
Merged

Update TensorFlow backend for Python 3.12 compatibility (Keras 2 via tf-keras) - LTS DLC 2 #3091

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
a9079c0
Relax tables
maximpavliv Sep 9, 2025
383e5e0
Relax pyside6
maximpavliv Sep 9, 2025
f5ca667
Relax tensorflow non-Windows
maximpavliv Sep 9, 2025
93bd9b5
Relax tensorflow-macos
maximpavliv Sep 9, 2025
c807975
Add _tf_legacy.py
maximpavliv Sep 9, 2025
77e62ed
Relax tables in requirements.txt
maximpavliv Sep 9, 2025
e7b0088
Relax tensorflow in requirements.txt
maximpavliv Sep 9, 2025
9776656
Add python 3.11 and 3.12 to testing CI/CD
maximpavliv Sep 9, 2025
504b556
CircleCI - use python 3.12
maximpavliv Sep 9, 2025
c9eddf2
Update torch pinned version
maximpavliv Sep 9, 2025
2649ead
Update Github actions trigger events
maximpavliv Sep 9, 2025
fe1da02
Add os condition for intel-openmp in requirements.txt
maximpavliv Sep 9, 2025
95880f9
Relax Matplotlib
maximpavliv Sep 9, 2025
8083b72
Remove mobilenet from functional tests
maximpavliv Sep 9, 2025
49f8d60
Random sample from list
maximpavliv Sep 9, 2025
de31770
Move tensorflow import to pose_estimation_tensorflow
maximpavliv Sep 9, 2025
1f3b1d9
Update 3mice and reid colab installations
maximpavliv Sep 9, 2025
8b5eaec
Update modelzoo colab installation
maximpavliv Sep 9, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update 3mice and reid colab installations
  • Loading branch information
@maximpavliv
maximpavliv committed Sep 9, 2025
commit 1f3b1d91ea599583487f901a34feac9a8fe9530b
42 changes: 3 additions & 39 deletions examples/COLAB/COLAB_3miceDemo.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,42 +32,7 @@
"- a quick guide to maDLC: https://deeplabcut.github.io/DeepLabCut/docs/tutorial.html\n",
"- a demo COLAB for how to use maDLC on your own data: https://github.com/DeepLabCut/DeepLabCut/blob/main/examples/COLAB/COLAB_maDLC_TrainNetwork_VideoAnalysis.ipynb\n",
"\n",
"### To get started, please go to \"Runtime\" ->\"change runtime type\"->select \"Python3\", and then select \"GPU\"\n",
"\n",
"As the COLAB environments were updated to CUDA 12.X and Python 3.11, we need to install DeepLabCut and TensorFlow in a distinct way to get TensorFlow to connect to the GPU."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"id": "HoNN2_0Z9rr_"
},
"outputs": [],
"source": [
"# Install TensorFlow, tensorpack and tf_slim versions compatible with DeepLabCut\n",
"!pip install \"tensorflow==2.12.1\" \"tensorpack>=0.11\" \"tf_slim>=1.1.0\""
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Downgrade PyTorch to a version using CUDA 11.8 and cudnn 8\n",
"# This will also install the required CUDA libraries, for both PyTorch and TensorFlow\n",
"!pip install torch==2.3.1 torchvision --index-url https://download.pytorch.org/whl/cu118"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Install the latest version of DeepLabCut\n",
"!pip install \"git+https://github.com/DeepLabCut/DeepLabCut.git\""
"### To get started, please go to \"Runtime\" ->\"change runtime type\"->select \"Python3\", and then select \"GPU\""
]
},
{
Expand All @@ -76,9 +41,8 @@
"metadata": {},
"outputs": [],
"source": [
"# As described in https://www.tensorflow.org/install/pip#step-by-step_instructions, \n",
"# create symbolic links to NVIDIA shared libraries:\n",
"!ln -svf /usr/local/lib/python3.11/dist-packages/nvidia/*/lib/*.so* /usr/local/lib/python3.11/dist-packages/tensorflow"
"# Install DeepLabCut 2.X (LTS)\n",
"!pip install \"deeplabcut[tf]\""
]
},
{
Expand Down
40 changes: 3 additions & 37 deletions examples/COLAB/COLAB_transformer_reID.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,40 +32,7 @@
"- a quick guide to maDLC: https://deeplabcut.github.io/DeepLabCut/docs/tutorial.html\n",
"- a demo COLAB for how to use maDLC on your own data: https://github.com/DeepLabCut/DeepLabCut/blob/main/examples/COLAB/COLAB_maDLC_TrainNetwork_VideoAnalysis.ipynb\n",
"\n",
"### To get started, please go to \"Runtime\" ->\"change runtime type\"->select \"Python3\", and then select \"GPU\"\n",
"\n",
"As the COLAB environments were updated to CUDA 12.X and Python 3.11, we need to install DeepLabCut and TensorFlow in a distinct way to get TensorFlow to connect to the GPU."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Install TensorFlow, tensorpack and tf_slim versions compatible with DeepLabCut\n",
"!pip install \"tensorflow==2.12.1\" \"tensorpack>=0.11\" \"tf_slim>=1.1.0\""
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Downgrade PyTorch to a version using CUDA 11.8 and cudnn 8\n",
"# This will also install the required CUDA libraries, for both PyTorch and TensorFlow\n",
"!pip install torch==2.3.1 torchvision --index-url https://download.pytorch.org/whl/cu118"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Install the latest version of DeepLabCut\n",
"!pip install \"git+https://github.com/DeepLabCut/DeepLabCut.git\""
"### To get started, please go to \"Runtime\" ->\"change runtime type\"->select \"Python3\", and then select \"GPU\""
]
},
{
Expand All @@ -74,9 +41,8 @@
"metadata": {},
"outputs": [],
"source": [
"# As described in https://www.tensorflow.org/install/pip#step-by-step_instructions, \n",
"# create symbolic links to NVIDIA shared libraries:\n",
"!ln -svf /usr/local/lib/python3.11/dist-packages/nvidia/*/lib/*.so* /usr/local/lib/python3.11/dist-packages/tensorflow"
"# Install DeepLabCut 2.X (LTS)\n",
"!pip install \"deeplabcut[tf]\""
]
},
{
Expand Down