diff --git a/requirements.txt b/requirements.txt
index 8acb9e8..22dee09 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -79,6 +79,9 @@ tensorflow-estimator==2.5.0rc0
 termcolor==1.1.0
 text-unidecode==1.3
 threadpoolctl==2.1.0
+torch==1.8.1
+torchaudio==0.8.1
+torchvision==0.9.1
 tqdm==4.59.0
 typing-extensions==3.7.4.3
 urllib3==1.26.4