Added torch requirement to setup.py (#82)

e874ef04 · Charles Coulombe · Soumith Chintala · e868060d · e874ef04
Commit e874ef04 authored Feb 12, 2019 by Charles Coulombe Committed by Soumith Chintala Feb 12, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

setup.py setup.py +3 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -36,4 +36,6 @@ setup(
            extra_compile_args=eca,
            extra_link_args=ela),
    ],
-    cmdclass={'build_ext': BuildExtension})
+    cmdclass={'build_ext': BuildExtension},
+    install_requires=['torch']
+)