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`
|
2020-08-04 15:56:36 +08:00
|
|
|
- gdown (for automatically downloading pre-train model)
|
|
|
|
- sklearn
|
|
|
|
- termcolor
|
|
|
|
- tabulate
|
2020-08-12 16:27:57 +08:00
|
|
|
- [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
|
2021-02-23 13:50:16 +08:00
|
|
|
pip install -r docs/requirements.txt
|
2020-09-08 11:14:05 +08:00
|
|
|
```
|
|
|
|
|
|
|
|
# Set up with Dockder
|
|
|
|
comming soon
|