nvidia-ml-py3>=7.352.0,<12

Signed-off-by: miguelgfierro <[email protected]>
recommenders-team · Nov 24, 2023 · 8daaf26 · 8daaf26
1 parent 8c6aaed
commit 8daaf26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -57,7 +57,7 @@
 # shared dependencies
 extras_require = {
     "gpu": [
-        "nvidia-ml-py3>=7.352.0",
+        "nvidia-ml-py3>=7.352.0,<12",
         "tensorflow==2.8.4",  # FIXME: Temporarily pinned due to issue with TF version > 2.10.1 See #2018
         "tf-slim>=1.1.0",
         "torch>=1.13.1",  # for CUDA 11 support