fast-reid/INSTALL.md

30 lines
827 B
Markdown
Raw Permalink Normal View History

2020-05-28 20:01:20 +08:00
# Installation
## Requirements
- Linux or macOS with python ≥ 3.6
2020-08-21 11:20:00 +08:00
- PyTorch ≥ 1.6
2020-06-02 10:19:20 +08:00
- torchvision that matches the Pytorch installation. You can install them together at [pytorch.org](https://pytorch.org/) to make sure of this.
2020-05-28 20:01:20 +08:00
- [yacs](https://github.com/rbgirshick/yacs)
2020-06-02 10:19:20 +08:00
- Cython (optional to compile evaluation code)
- tensorboard (needed for visualization): `pip install tensorboard`
- gdown (for automatically downloading pre-train model)
- sklearn
- termcolor
- tabulate
- [faiss](https://github.com/facebookresearch/faiss) `pip install faiss-cpu`
2020-09-08 11:14:05 +08:00
# Set up with Conda
```shell script
conda create -n fastreid python=3.7
conda activate fastreid
conda install pytorch==1.6.0 torchvision tensorboard -c pytorch
pip install -r docs/requirements.txt
2020-09-08 11:14:05 +08:00
```
# Set up with Dockder
Please check the [docker folder](docker)