115 lines
4.9 KiB
Python
Executable File
115 lines
4.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Copyright The PyTorch Lightning team.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import os
|
|
from importlib.util import module_from_spec, spec_from_file_location
|
|
|
|
from setuptools import find_packages, setup
|
|
|
|
# https://packaging.python.org/guides/single-sourcing-package-version/
|
|
# http://blog.ionelmc.ro/2014/05/25/python-packaging/
|
|
_PATH_ROOT = os.path.dirname(__file__)
|
|
_PATH_REQUIRE = os.path.join(_PATH_ROOT, "requirements")
|
|
|
|
|
|
def _load_py_module(fname, pkg="pytorch_lightning"):
|
|
spec = spec_from_file_location(os.path.join(pkg, fname), os.path.join(_PATH_ROOT, pkg, fname))
|
|
py = module_from_spec(spec)
|
|
spec.loader.exec_module(py)
|
|
return py
|
|
|
|
|
|
about = _load_py_module("__about__.py")
|
|
setup_tools = _load_py_module("setup_tools.py")
|
|
|
|
# https://setuptools.readthedocs.io/en/latest/setuptools.html#declaring-extras
|
|
# Define package extras. These are only installed if you specify them.
|
|
# From remote, use like `pip install pytorch-lightning[dev, docs]`
|
|
# From local copy of repo, use like `pip install ".[dev, docs]"`
|
|
extras = {
|
|
# 'docs': load_requirements(file_name='docs.txt'),
|
|
"examples": setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name="examples.txt"),
|
|
"loggers": setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name="loggers.txt"),
|
|
"extra": setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name="extra.txt"),
|
|
"test": setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name="test.txt"),
|
|
}
|
|
extras["dev"] = extras["extra"] + extras["loggers"] + extras["test"]
|
|
extras["all"] = extras["dev"] + extras["examples"] # + extras['docs']
|
|
|
|
# These packages shall be installed only on GPU machines
|
|
PACKAGES_GPU_ONLY = ["horovod"]
|
|
# create a version for CPU machines
|
|
for ex in ("cpu", "cpu-extra"):
|
|
kw = ex.split("-")[1] if "-" in ex else "all"
|
|
# filter cpu only packages
|
|
extras[ex] = [pkg for pkg in extras[kw] if not any(pgpu.lower() in pkg.lower() for pgpu in PACKAGES_GPU_ONLY)]
|
|
|
|
long_description = setup_tools._load_readme_description(
|
|
_PATH_ROOT, homepage=about.__homepage__, version=about.__version__
|
|
)
|
|
|
|
# https://packaging.python.org/discussions/install-requires-vs-requirements /
|
|
# keep the meta-data here for simplicity in reading this file... it's not obvious
|
|
# what happens and to non-engineers they won't know to look in init ...
|
|
# the goal of the project is simplicity for researchers, don't want to add too much
|
|
# engineer specific practices
|
|
setup(
|
|
name="pytorch-lightning",
|
|
version=about.__version__,
|
|
description=about.__docs__,
|
|
author=about.__author__,
|
|
author_email=about.__author_email__,
|
|
url=about.__homepage__,
|
|
download_url="https://github.com/PyTorchLightning/pytorch-lightning",
|
|
license=about.__license__,
|
|
packages=find_packages(exclude=["tests", "tests/*", "benchmarks", "legacy", "legacy/*"]),
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
include_package_data=True,
|
|
zip_safe=False,
|
|
keywords=["deep learning", "pytorch", "AI"],
|
|
python_requires=">=3.6",
|
|
setup_requires=[],
|
|
install_requires=setup_tools._load_requirements(_PATH_ROOT),
|
|
extras_require=extras,
|
|
project_urls={
|
|
"Bug Tracker": "https://github.com/PyTorchLightning/pytorch-lightning/issues",
|
|
"Documentation": "https://pytorch-lightning.rtfd.io/en/latest/",
|
|
"Source Code": "https://github.com/PyTorchLightning/pytorch-lightning",
|
|
},
|
|
classifiers=[
|
|
"Environment :: Console",
|
|
"Natural Language :: English",
|
|
# How mature is this project? Common values are
|
|
# 3 - Alpha, 4 - Beta, 5 - Production/Stable
|
|
"Development Status :: 4 - Beta",
|
|
# Indicate who your project is intended for
|
|
"Intended Audience :: Developers",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"Topic :: Scientific/Engineering :: Image Recognition",
|
|
"Topic :: Scientific/Engineering :: Information Analysis",
|
|
# Pick your license as you wish
|
|
"License :: OSI Approved :: Apache Software License",
|
|
"Operating System :: OS Independent",
|
|
# Specify the Python versions you support here. In particular, ensure
|
|
# that you indicate whether you support Python 2, Python 3 or both.
|
|
"Programming Language :: Python :: 3",
|
|
"Programming Language :: Python :: 3.6",
|
|
"Programming Language :: Python :: 3.7",
|
|
"Programming Language :: Python :: 3.8",
|
|
"Programming Language :: Python :: 3.9",
|
|
],
|
|
)
|