INSTALL.md

Installation

Contents

• Requirements

• Install from a local clone

Requirements

• Python >= 3.7 (Recommend to use Anaconda or Miniconda)
• PyTorch >= 1.7
• NVIDIA GPU + CUDA
• Linux (We have not tested on Windows)

Install from a local clone

1. Clone the repo

   git clone https://github.com/DianaNerualNetwork/SegResearchToolkit

2. Install dependent packages

   cd SegResearchToolkit
   conda create -n seg python=3.8
   conda activate seg
   # please note your cudann version
   #CUDA 10.2
   conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
   # CUDA 11.3
   conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
   # pip install others 
   pip install opencv-python==4.6.0.66 pytorch_lightning==1.9.3 tensorboard==2.11.2 scikit-image scikit-learn
   pip install PyYAML filelock matplotlib