Add dependencies to setup

This way dependencies are automatically installed with the package.
richzhang · Jan 29, 2021 · b2ca903 · b2ca903
1 parent 05f4ec2
commit b2ca903
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -14,6 +14,7 @@
  packages=['lpips'],
  package_data={'lpips': ['weights/v0.0/*.pth','weights/v0.1/*.pth']},
  include_package_data=True,
+ install_requires=["torch>=0.4.0", "torchvision>=0.2.1", "numpy>=1.14.3", "scipy>=1.0.1", "tqdm>=4.28.1"],
  classifiers=[
  "Programming Language :: Python :: 3",
  "License :: OSI Approved :: BSD License",