bumbp pyro and torch versions

gelles-brandeis · Jul 7, 2023 · a82eb48 · a82eb48
1 parent 318e8fc
commit a82eb48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -70,12 +70,12 @@
         "pandas",
         "protobuf>=3.9,<3.20",
         "pykeops>=2.0",
-        "pyro-ppl==1.8.2",
+        "pyro-ppl>=1.8.5",
         "pyyaml>=6.0",
         "scikit-learn",
         "scipy",
         "tensorboard",
-        "torch>=1.11.0",
+        "torch>=2.0.1",
         "typer",
     ],
     extras_require={