Merge pull request #26 from instadeepai/versions/update_tf_versions

Changed versions after successful install on Dell XPS
instadeepai · Apr 17, 2024 · 2dcb0d1 · 2dcb0d1
2 parents 8b7fc9c + 5a7612c
commit 2dcb0d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -28,9 +28,9 @@
     install_requires=[
         "numpy",
         "dm_tree",
-        "tensorflow[and-cuda]==2.16.1.*",
+        "tensorflow[and-cuda]==2.15.1.*",
         "tensorflow_io",
-        "tensorflow-probability[tf]==0.24.0",
+        "tensorflow-probability[tf]==0.23.0",
         "dm_sonnet",
         "wandb",
         "absl-py",