Browse Source

update setup info (#233)

pull/394/head
ver217 3 years ago committed by Frank Lee
parent
commit
24f8583cc4
  1. 32
      setup.py
  2. 1
      version.txt

32
setup.py

@ -86,6 +86,16 @@ def fetch_requirements(path):
return [r.strip() for r in fd.readlines()] return [r.strip() for r in fd.readlines()]
def fetch_readme():
with open('README.md', encoding='utf-8') as f:
return f.read()
def get_version():
with open('version.txt') as f:
return f.read().strip()
if build_cuda_ext: if build_cuda_ext:
try: try:
import torch import torch
@ -177,7 +187,7 @@ if build_cuda_ext:
setup( setup(
name='colossalai', name='colossalai',
version='0.0.2', version=get_version(),
packages=find_packages(exclude=('benchmark', packages=find_packages(exclude=('benchmark',
'docker', 'docker',
'tests', 'tests',
@ -188,10 +198,28 @@ setup(
'requirements', 'requirements',
'*.egg-info',)), '*.egg-info',)),
description='An integrated large-scale model training system with efficient parallelization techniques', description='An integrated large-scale model training system with efficient parallelization techniques',
long_description=fetch_readme(),
long_description_content_type='text/markdown',
url='https://www.colossalai.org',
project_urls={
'Forum': 'https://github.com/hpcaitech/ColossalAI/discussions',
'Bug Tracker': 'https://github.com/hpcaitech/ColossalAI/issues',
'Examples': 'https://github.com/hpcaitech/ColossalAI-Examples',
'Documentation': 'http://colossalai.readthedocs.io',
'Github': 'https://github.com/hpcaitech/ColossalAI',
},
ext_modules=ext_modules, ext_modules=ext_modules,
cmdclass={'build_ext': BuildExtension} if ext_modules else {}, cmdclass={'build_ext': BuildExtension} if ext_modules else {},
install_requires=fetch_requirements('requirements/requirements.txt'), install_requires=fetch_requirements('requirements/requirements.txt'),
extras_require={ extras_require={
'zero': fetch_requirements('requirements/requirements-zero.txt'), 'zero': fetch_requirements('requirements/requirements-zero.txt'),
} },
python_requires='>=3.7',
classifiers=[
'Programming Language :: Python :: 3',
'License :: OSI Approved :: Apache Software License',
'Environment :: GPU :: NVIDIA CUDA',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Topic :: System :: Distributed Computing',
],
) )

1
version.txt

@ -0,0 +1 @@
0.0.2
Loading…
Cancel
Save