-
Notifications
You must be signed in to change notification settings - Fork 44
/
setup.py
17 lines (16 loc) · 782 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from setuptools import setup
from setuptools import find_packages
setup(name='sparktorch',
version='0.2.0',
description='Distributed training of PyTorch networks on Apache Spark with ML Pipeline support',
keywords=['pytorch', 'spark', 'sparktorch', 'machine learning', 'deep learning'],
url='https://github.com/dmmiller612/sparktorch',
download_url='https://github.com/dmmiller612/sparktorch/archive/0.2.0.tar.gz',
author='Derek Miller',
author_email='dmmiller612@gmail.com',
long_description=open("README.md", "r", encoding='utf-8').read(),
long_description_content_type="text/markdown",
install_requires=['torch', 'flask', 'requests', 'dill'],
license='MIT',
packages=find_packages(),
zip_safe=False)