Skip to content

Latest commit

 

History

History
28 lines (20 loc) · 1.11 KB

INSTALL.md

File metadata and controls

28 lines (20 loc) · 1.11 KB

Installation

The installation follows Detectron2. Here we provide a quickstart guide, and refer to the solutions from Detectron2 should any issue arise.

Requirements

  • Linux or macOS with Python ≥ 3.6
  • PyTorch ≥ 1.6 and torchvision that matches the PyTorch installation
  • OpenCV is optional for training and inference, yet is needed by our demo and visualization

Quick Start

# environment
conda create -n regionclip python=3.9
source activate regionclip
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

# RegionCLIP
git clone [email protected]:microsoft/RegionCLIP.git
python -m pip install -e RegionCLIP

# other dependencies
pip install opencv-python timm diffdist h5py sklearn ftfy
pip install git+https://github.com/lvis-dataset/lvis-api.git

To rebuild, use rm -rf build/ **/*.so to clean the old build first. You often need to rebuild detectron2 after reinstalling PyTorch.