diff --git a/requirements.txt b/requirements.txt
index 53f5496..7057ca9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,10 +1,10 @@
-# need to install torch==1.13.1+cu117 separately with  `pip install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117`
+# torch>=1.13.1 should be installed separately with specific index if CUDA support needed: `pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118`
 matplotlib
 numpy==1.26.4
 opencv-python
 mmcv==1.4.7
 yapf==0.32.0
-pytorch-lightning==1.7.7
+pytorch-lightning>=1.8.6,<2.0.0
 pytest
 setuptools>=59.5.0
 wandb
@@ -12,7 +12,7 @@ plotly
 scipy
 gradio
 einops
-pydantic==1.10
+pydantic>=1.10,<2.0.0
 fire
 packaging
-torchmetrics==0.11.4
\ No newline at end of file
+torchmetrics>=0.11.4,<1.0.0
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 41581bc..08ee550 100644
--- a/setup.py
+++ b/setup.py
@@ -10,19 +10,19 @@
     packages=["risk_biased"],
     zip_safe=False,
     install_requires=[
-        "torch==1.13.1+cu117",
+        "torch>=1.13.1",
         "matplotlib",
         "numpy==1.26.4",
         "mmcv==1.4.7",
-        "pytorch-lightning==1.7.7",
+        "pytorch-lightning>=1.8.6,<2.0.0",
         "pytest",
         "setuptools>=59.5.0",
         "wandb",
         "plotly",
         "scipy",
-        "pydantic==1.10",
+        "pydantic>=1.10,<2.0.0",
         "gradio",
         "einops",
-        "torchmetrics==0.11.4"
+        "torchmetrics>=0.11.4,<1.0.0"
     ],
 )