diff --git a/setup.py b/setup.py
index dde39a7..2772db1 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,7 @@
     author='CNChTu',
     author_email='2921046558@qq.com',
     url='https://github.com/CNChTu/FCPE',
-    install_requires=['einops', 'local_attention', 'torchaudio', 'numpy'],
+    install_requires=['einops', 'local_attention', 'torch', 'torchaudio', 'numpy'],
     packages=['torchfcpe'],
     package_data={'torchfcpe': ['assets/*']},
     long_description=long_description,