Newer
Older
import setuptools
with open("README.md", "r", encoding="utf-8") as fh:
long_description = fh.read()
setuptools.setup(
name="predtuner",
version="0.1",
author="Yifan Zhao",
author_email="yifanz16@illinois.edu",
description="A package for predictive and empirical approximation autotuning",
long_description=long_description,
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"
]