diff --git a/requirements.txt b/requirements.txt
index 2a24bbd..a00bdc4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,8 +4,7 @@ matplotlib==3.3.1
 numpy==1.21.6
 scipy==1.5.2
 tensorboard==2.3.0
-torch==1.6.0
-torchvision==0.7.0
+torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
 unidecode==1.3.4
 pyopenjtalk==0.2.0
 jamo==0.4.1