File size: 816 Bytes
2c58401 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
# Installation
We provide the instructions to install the dependency packages.
## Requirements
We test the code in the following environments, other versions may also be compatible:
- CUDA 11.1
- Python 3.7
- Pytorch 1.8.1
## Setup
First, clone the repository locally.
```
git clone https://github.com/wjn922/ReferFormer.git
```
Then, install Pytorch 1.8.1 using the conda environment.
```
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 -c pytorch
```
Install the necessary packages and 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 ../..
``` |