Skip to content

Latest commit

 

History

History
37 lines (28 loc) · 1011 Bytes

INSTALL.md

File metadata and controls

37 lines (28 loc) · 1011 Bytes
Raw

  1. First clone the project and install requirments

    git clone https://github.com/cfzd/FcaNet.git
cd FcaNet
pip install -r requirements.txt

  2. Install Nvidia DALI

    For CUDA 10

    pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda100

    For CUDA 11

    pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda110

  3. Install Nvidia APEX (Optional, only for mixed precision training)

    git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

  4. Install mmdetection (Optional, only for detection and instance segmentation models).

    pip install mmcv-full==1.1.4
cd FcaNet/mmdetection
pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"