Skip to content

Latest commit

 

History

History
26 lines (19 loc) · 579 Bytes

install.md

File metadata and controls

26 lines (19 loc) · 579 Bytes

Installation

We provide the instructions to install the dependency packages.

Setup

First, clone the repository locally.

git clone https://github.com/bo-miao/SgMg

Then, install Pytorch==1.11.0 (CUDA 11.3) torchvision==0.12.0 and the necessary packages as well as pycocotools.

pip install -r requirements.txt 
pip install 'git+https://github.com/facebookresearch/fvcore' 
pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

Finally, compile CUDA operators.

cd models/ops
python setup.py build install
cd ../..