diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..6eed2b8
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,22 @@
+--extra-index-url https://download.pytorch.org/whl/cu118 
+torch==2.3.0 
+torchvision==0.18.0 
+torchaudio==2.3.0 
+
+numpy==1.26.4
+pyyaml==6.0.1
+opencv-python==4.10.0.84
+scipy==1.13.1
+imageio==2.34.2
+lmdb==1.4.1
+tqdm==4.66.4
+timm==1.0.7
+rich==13.7.1
+ffmpeg==1.4
+onnxruntime-gpu==1.18.0
+onnx==1.16.1
+scikit-image==0.24.0
+albumentations==1.4.10
+matplotlib==3.9.0
+imageio-ffmpeg==0.5.1
+tyro==0.8.5