diff --git a/setup.py b/setup.py
index 0ce7b4c..8caf58b 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,7 @@ setup_info = dict(
     install_requires=[
         'numpy', 'nltk', 'inflect', 'librosa', 'unidecode', 'numba', 
         'tqdm', 'matplotlib', 'tensorboardX', 'tensorboard', 'scipy',
-        'ruamel.yaml', 'pandas', 'sox', 'paddlepaddle', 
+        'ruamel.yaml', 'pandas', 'sox', 'paddlepaddle-gpu', 
     ],
 
     # Package info