Skip to content

Latest commit

 

History

History
executable file
·
44 lines (32 loc) · 1.06 KB

INSTALL.md

File metadata and controls

executable file
·
44 lines (32 loc) · 1.06 KB

Installation

Requirements

  • Linux
  • Python 3.6+
  • PyTorch 1.3+
  • CUDA 9.2+ (If you build PyTorch from source, CUDA 9.0 is also compatible)
  • NCCL 2+
  • GCC 5+
  • mmcv = 1.6.1
  • mmdetection = 2.25.1

Install QDTrack

a. Create a conda virtual environment and activate it.

conda create -n qdtrack python=3.7 -y
conda activate qdtrack

b. Install PyTorch and torchvision following the official instructions, e.g.,

conda install pytorch torchvision -c pytorch

Note: Make sure that your compilation CUDA version and runtime CUDA version match. You can check the supported CUDA version for precompiled packages on the PyTorch website.

c. Install mmcv and mmdetection.

pip install mmcv-full==1.6.1
pip install mmdet==2.25.1

You can also refer to the offical instructions.

d. Install QDTrack

python setup.py develop