Issue with TF

Signed-off-by: miguelgfierro <[email protected]>
recommenders-team · Oct 12, 2023 · 93ed1c0 · 93ed1c0
1 parent d3da0fe
commit 93ed1c0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -60,9 +60,7 @@
 extras_require = {
     "gpu": [
         "nvidia-ml-py3>=7.352.0",
-        # TensorFlow compiled with CUDA 11.2, CuDNN 8.1
-        "tensorflow~=2.6.1;python_version=='3.6'",
-        "tensorflow~=2.7.0;python_version>='3.7'",
+        "tensorflow>=2.10.1",
         "tf-slim>=1.1.0",
         "torch>=1.13.1",  # for CUDA 11 support
         "fastai>=1.0.46,<2",

diff --git a/tests/security/test_dependency_security.py b/tests/security/test_dependency_security.py
@@ -33,7 +33,8 @@ def test_pandas():
 def test_tensorflow():
     # Security issue: https://github.com/advisories/GHSA-w5gh-2wr2-pm6g
     # Security issue: https://github.com/advisories/GHSA-r6jx-9g48-2r5r
-    assert tf.__version__ >= "2.5.1"
+    # Security issue: https://github.com/advisories/GHSA-xxcj-rhqg-m46g
+    assert tf.__version__ >= "2.10.1"
 
 
 @pytest.mark.gpu