Skip to content

Latest commit

 

History

History
36 lines (30 loc) · 1.11 KB

install.md

File metadata and controls

36 lines (30 loc) · 1.11 KB

Install

Requirements

We test the codes in the following environments, other versions may also be compatible but Pytorch vision should be >= 1.7

  • CUDA 11.3
  • Python 3.7
  • Pytorch 1.10.0
  • Torchvison 0.11.1

Install environment for Unicorn

# Pytorch and Torchvision
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

# YOLOX and some other packages
pip3 install -U pip && pip3 install -r requirements.txt
pip3 install -v -e .  # or python3 setup.py develop

# Install Deformable Attention
cd unicorn/models/ops
bash make.sh
cd ../../..

# Install mmcv, mmdet, bdd100k
cd external/qdtrack
wget -c https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/mmcv_full-1.4.6-cp37-cp37m-manylinux1_x86_64.whl # This should change according to cuda version and pytorch version
pip3 install --user mmcv_full-1.4.6-cp37-cp37m-manylinux1_x86_64.whl
pip3 install --user mmdet
git clone https://github.com/bdd100k/bdd100k.git
cd bdd100k
python3 setup.py develop --user
pip3 uninstall -y scalabel
pip3 install --user git+https://github.com/scalabel/scalabel.git
cd ../../..