Skip to content

Latest commit

 

History

History
56 lines (44 loc) · 1.46 KB

INSTALL.md

File metadata and controls

56 lines (44 loc) · 1.46 KB

Installation

We provide instructions to install the required dependencies.

Requirements:

  • python>=3.7
  • pytorch==1.8 (should work with pytorch >=1.8 as well but not tested)
  1. Unzip the repo and set the root directory:

    export ROOT=$(pwd)
    
  2. To use the same environment you can use conda and the environment file vstates_env.yml file provided. Please refer to Miniconda for details on installing conda.

    MINICONDA_ROOT=[to your Miniconda/Anaconda root directory]
    conda env create -f vstates_env.yml --prefix $MINICONDA_ROOT/envs/vstates
    conda activate vstates
    
  3. Install submodules:

    • Install Detectron2 (needed for SlowFast). If you have CUDA 11.1 and Pytorch 1.8 you can use:
    python -m pip install detectron2 -f \
    https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.8/index.html
    

    Please refer to detectron2 repository for more details.

    • Slowfast:
    cd $ROOT/submodules/SlowFast
    python setup.py build develop
    
    • Fairseq:
    cd $ROOT/submodules/fairseq
    pip install -e .
    
    • cocoapi:
    cd $ROOT/submodules/cocoapi/PythonAPI
    make
    
    • coco-caption: (NOTE: You may need to install java). No additional steps are needed.

    • coval:

    cd $ROOT/submodules/coval
    pip install -e .