diff --git a/.gitignore b/.gitignore
index 99efc91..6440e67 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,7 +5,7 @@ __pycache__/
 
 # C extensions
 *.so
-
+.idea
 # Distribution / packaging
 .Python
 build/
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..3c44c85
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,13 @@
+cudatoolkit=10.2.89
+cycler=0.10.0
+einops=0.3.0
+h5py=3.2.1
+matplotlib=3.4.2
+numpy=1.20.2
+numpy-base=1.20.2
+pytorch=1.8.1
+pyyaml=5.4.1
+scikit-learn=0.24.2
+scipy=1.6.3
+torchvision=0.9.1
+tqdm=4.61.1