setup.py

#!/usr/bin/env python

from setuptools import setup, find_packages

setup(
    name="prostate_cancer_segmentation",
    version="0.1.0",
    description="Semantic Segmentation on the PANDA dataset using Pytorch Lightning",
    author="john doe",
    author_email="vpeopleonatank@gmail.com",
    # REPLACE WITH YOUR OWN GITHUB PROJECT LINK
    url="https://github.com/vpeopleonatank/prostate_cancer_segmentation",
    python_requires=">=3.7.7",
    install_requires=[
        # Install torch first, depending on cuda version
        # "torch==1.7.1",
        # "torchvision==0.8.2",
        "pytorch-lightning==1.1.2",
        "gdown==3.12.2",
        "albumentations==0.5.2",
        "opencv-python==4.4.0.44",
        "hydra-core==1.0.4",
        "wandb==0.10.12",
        "pydantic==1.7.3",
    ],
    packages=find_packages(),
)