Our installation is simple. We provide pre-built binaries (pip, conda) and also easy instructions for building from source (pip, conda).


At a high level, project requires following system dependencies.

  • Linux

  • Python>=3.6.2 and <3.9

  • PyTorch>=1.4, PyTorch<=3.9.1

  • torchvision (matching PyTorch install)

  • CUDA (must be a version supported by the pytorch version)

  • OpenCV (optional)

Installing VISSL from pre-built binaries

Install VISSL conda package

This assumes you have conda 10.2.

conda create -n vissl python=3.8
conda activate vissl
conda install -c pytorch pytorch=1.7.1 torchvision cudatoolkit=10.2
conda install -c vissl -c iopath -c conda-forge -c pytorch -c defaults apex vissl

For all other versions of PyTorch, Python, CUDA, please modify the above instructions with the desired version. VISSL provides Apex packages for all combinations of pytorch, python and compatible cuda.

Install VISSL pip package

This example is with pytorch 1.5.1 and cuda 10.1. Please modify the PyTorch version, cuda version and matching apex version below for the desired settings.

  • Step 1: Create Virtual environment (pip)

python3 -m venv ~/venv
. ~/venv/bin/activate
  • Step 2: Install PyTorch, OpenCV and APEX (pip)

    • We use PyTorch=1.5.1 with CUDA 10.1 in the following instruction (user can chose their desired version).

    • There are several ways to install opencv, one possibility is as follows.

    • For APEX, we provide pre-built binary built with optimized C++/CUDA extensions provided by APEX.

pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f
pip install opencv-python
pip install -f apex

Note that, for the APEX install, you need to get the versions of CUDA, PyTorch, and Python correct in the URL. We provide APEX versions with all possible combinations of Python, PyTorch, CUDA. Select the right APEX Wheels if you desire a different combination.

On Google Colab, everything until this point is already set up. You install APEX there as follows.

import sys
import torch
!pip install -f{version_str}/download.html apex
  • Step 3: Install VISSL

pip install vissl
# verify installation
python -c 'import vissl'

That’s it! You are now ready to use Vissl.

  • Optional: Install Apex from source (common for both pip and conda)

Apex installation requires that you have a latest nvcc so the c++ extensions can be compiled with latest gcc (>=7.4). Check the APEX website for more instructions.

# see
# to select cuda architecture you want to build
CUDA_VER=10.1 TORCH_CUDA_ARCH_LIST="5.0;5.2;5.3;6.0;6.1;6.2;7.0;7.5" ./docker/common/