diff --git a/setup.py b/setup.py
index 366b0d9..c07b606 100644
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,7 @@ def get_description():
       license="Apache-2.0",
       include_package_data=True,
       packages=find_packages(),
-      install_requires=["tensorflow-gpu==1.15.2",
+      install_requires=["tensorflow_gpu==1.15.2",
                         "tensorboard",
                         "keras==2.3.1",
                         "py2bit==0.3.0",