42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
"""Install Compacter."""
|
||
|
|
import os
|
||
|
|
import setuptools
|
||
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
||
|
|
|
||
|
|
#os.environ['TORCH_CUDA_ARCH_LIST']="3.5;3.7;6.1;7.0;7.5;8.6+PTX"
|
||
|
|
|
||
|
|
def setup_package():
|
||
|
|
long_description = "examples_seq2seq"
|
||
|
|
setuptools.setup(
|
||
|
|
name='examples_seq2seq',
|
||
|
|
version='0.0.1',
|
||
|
|
description='seq2seq example',
|
||
|
|
long_description=long_description,
|
||
|
|
long_description_content_type='text/markdown',
|
||
|
|
author='Shengding Hu',
|
||
|
|
license='MIT License',
|
||
|
|
packages=setuptools.find_packages(
|
||
|
|
exclude=['docs', 'tests', 'scripts']),
|
||
|
|
dependency_links=[
|
||
|
|
'https://download.pytorch.org/whl/torch_stable.html',
|
||
|
|
],
|
||
|
|
classifiers=[
|
||
|
|
'Intended Audience :: Developers',
|
||
|
|
'Intended Audience :: Science/Research',
|
||
|
|
'License :: OSI Approved :: MIT License',
|
||
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
||
|
|
'Programming Language :: Python :: 3',
|
||
|
|
'Programming Language :: Python :: 3.7.10',
|
||
|
|
],
|
||
|
|
keywords='text nlp machinelearning',
|
||
|
|
cmdclass={"build_ext": BuildExtension},
|
||
|
|
install_requires=[
|
||
|
|
"pyarrow==7.0.0",
|
||
|
|
"datasets==1.17.0"
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
setup_package()
|