diff --git a/requirements.txt b/requirements.txt
index 0527b31df..f16720a7f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,11 +1,19 @@
-torch
-torchdiffeq
-torchsde
-einops
+torch~=2.0.0+cu118
+torchdiffeq~=0.2.3
+torchsde~=0.2.5
+einops~=0.6.0
 open-clip-torch
 transformers>=4.25.1
 safetensors>=0.3.0
 pytorch_lightning
-aiohttp
-accelerate
-pyyaml
+aiohttp~=3.8.4
+accelerate~=0.18.0
+pyyaml~=6.0
+opencv-python~=4.7.0.72
+numpy~=1.24.2
+tqdm~=4.65.0
+torchvision~=0.15.1
+timm~=0.6.13
+scipy~=1.10.1
+Pillow~=9.5.0
+psutil~=5.9.4
\ No newline at end of file