- Linux or macOS with python ≥ 3.6
- PyTorch ≥ 1.2
- torchvision that matches the Pytorch installation. You can install them together at pytorch.org to make sure of this.
- yacs
- Cython (optional to compile evaluation code)
- tensorboard (needed for visualization):
pip install tensorboard