diff --git a/requirements.txt b/requirements.txt index 6d7faf64b..f1e96f658 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,7 +14,7 @@ pytest scikit-image==0.18.3 scipy==1.7.1 shapely==1.6.4 -tensorflow-gpu==2.5.1 +tensorflow-gpu==2.12.0 torch==1.4.0 torchvision==0.5.0 ##### Tracking dependencies ##### diff --git a/setup.py b/setup.py index a64abcf1a..7c4e6bc52 100644 --- a/setup.py +++ b/setup.py @@ -27,7 +27,7 @@ "scikit-image==0.18.3", "scipy==1.7.1", "shapely==1.6.4", - "tensorflow-gpu==2.5.1", + "tensorflow-gpu==2.12.0", "torch==1.4.0", "torchvision==0.5.0", ##### Tracking dependencies #####