Skip to content

Latest commit

 

History

History
27 lines (21 loc) · 728 Bytes

INSTALL.md

File metadata and controls

27 lines (21 loc) · 728 Bytes
Raw

Installation

Our experiments are based on Python 3.8, PyTorch 1.9.0, and CUDA 11.1. Newer version of PyTorch may also work.

Installation Outline

  • Download and install Pytorch
  • Download and install Detectron2. We use the version at the commit ID cc87e7ec
  • Download and install GRiT

Installation Example

conda create --name grit python=3.8 -y
conda activate grit
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
git checkout cc87e7ec
pip install -e .

cd ..
git clone https://github.com/JialianW/GRiT.git
cd GRiT
pip install -r requirements.txt