diff --git a/dev_requirements.txt b/dev_requirements.txt
new file mode 100644
index 000000000..23cf9dac0
--- /dev/null
+++ b/dev_requirements.txt
@@ -0,0 +1,2 @@
+isort==4.3.21
+flake8
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 71da6fe6b..12b19d953 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,10 +10,10 @@ tb-nightly
 future
 yacs
 gdown
-flake8
 yapf
-isort==4.3.21
 imageio
+torch
+torchvision
 
 # Export --------------------------------------
 # onnx