Update setup.py

tensorflow/tensorflow#36347 (comment)
MiraldiLab · Feb 21, 2021 · 858748f · 858748f
1 parent 7d136c3
commit 858748f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",