diff --git a/env.yaml b/env.yaml deleted file mode 100644 index 735790b90eacedc2bb9bb1969c6af91a47907e5f..0000000000000000000000000000000000000000 --- a/env.yaml +++ /dev/null @@ -1,19 +0,0 @@ -name: predtuner -channels: - - pytorch - - defaults -dependencies: - - matplotlib=3.3.2 - - networkx=2.5 - - python=3.8.5 - - pytorch=1.7.0 - - torchvision=0.8.1 - - tqdm=4.50.2 - - pandas=1.1.3 - - pip=20.2.4 - - wheel=0.35.1 - - jsonpickle=1.5 - - pip: - - argparse - - opentuner==0.8.3 # Must be 0.8.3, they fixed an important bug - - sqlalchemy==1.3 diff --git a/setup.py b/setup.py index 8d08aa7a80b05d60f47a63b5bacb77a0d105fa61..87902831ed7cf1f869f1c4e5494b91c8e7e1c3ae 100644 --- a/setup.py +++ b/setup.py @@ -13,4 +13,17 @@ setuptools.setup( long_description_content_type="text/markdown", url="https://github.com/Evan-Zhao/predictive-tuner", packages=["predtuner"], + install_requires=[ + "matplotlib>=3.3", + "networkx>=2.5", + "torch==1.7.1", + "torchvision==0.8.2", + "tqdm>=4.50", + "pandas>=1.1", + "jsonpickle>=1.5", + "argparse>=1.4", + "wheel", # Use wheel to build the following: + "opentuner==0.8.3", # Must be 0.8.3, they fixed an important bug + "sqlalchemy==1.3" + ] )