Install

Requirements

Linux (Windows is not officially supported)
Python 3.8
PyTorch 1.8.2 or higher
torchvison 0.9.2 or higher

About apex，there has a bug in NVIDIA/apex, see How can I solve it？ #1215. Now i use pytorch naive apex util in ZCls2.

I recommend people to use docker training environment, see

• NGC Pytorch
• PyTorch Release Notes
• NVIDIA/apex

Install ZCls2

Source install

• Clone the ZCls2 repository.

git clone https://github.com/ZJCV/ZCls2.git
cd ZCls2

• Install ZCls2.

python3 setup.py install

Pip install (RECOMMEND)

pip3 install zcls2