Skip to content

Latest commit

 

History

History
29 lines (23 loc) · 994 Bytes

INSTALL.md

File metadata and controls

29 lines (23 loc) · 994 Bytes
Raw

Installation

Requirements

  • Linux or macOS with Python ≥ 3.6
  • PyTorch ≥ 1.8. Install them together at pytorch.org to make sure of this. Note, please check PyTorch version matches that is required by Detectron2.
  • Detectron2: follow Detectron2 installation instructions.

Example conda environment setup

conda create --name gtr python=3.8 -y
conda activate gtr
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia

# under your working directory
git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
pip install -e .

cd ..
git clone https://github.com/xingyizhou/GTR.git --recurse-submodules
cd GTR
pip install -r requirements.txt

Our project uses a submodule, CenterNet2. If you forget to add --recurse-submodules, do git submodule init and then git submodule update.