setup.py

from setuptools import find_packages, setup

setup(
    name="cell_segmentation",
    packages=find_packages(),
    version="0.1.0",
    description="sartorius cell instance segmentation kaggle competition",
    author="Leon Truong",
    license="MIT",
    install_requires=[
        "black",
        "click==8.0.3",
        "numpy==1.21.3",
        "pandas==1.3.4",
        "opencv-python==4.5.4.58",
        "scikit-image==0.18.3",
        "torch@https://download.pytorch.org/whl/cu113/torch-1.10.0%2Bcu113-cp37-cp37m-linux_x86_64.whl",
        "torchvision@https://download.pytorch.org/whl/cu113/torchvision-0.11.1%2Bcu113-cp37-cp37m-linux_x86_64.whl",
        "detectron2@https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/detectron2-0.6%2Bcu113-cp37-cp37m-linux_x86_64.whl",
    ],
    python_requires=">=3.7",
)