diff --git a/requirements.txt b/requirements.txt
index dd4249f..e6e57cd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,8 @@
+matplotlib==3.5.3
 numpy==1.21.2
 pandas==1.3.3
 pynvml==11.4.1
 reformer-pytorch==1.4.3
-torch==1.8.0
+scipy==1.7.3
+torch==1.9.1+cu111
 tqdm==4.62.2
\ No newline at end of file