Installation

Prerequisites

Condensa requires:

  • A working Linux installation (we use Ubuntu 18.04)

  • NVIDIA drivers and CUDA 10+ for GPU support

  • Python 3.5 or newer

  • PyTorch 1.0 or newer

Installation with pip

The most straightforward way of installing Condensa is via pip:

pip install condensa

Installation from Source

Retrieve the latest source code from the Condensa repository:

git clone https://github.com/NVlabs/condensa.git

Navigate to the source code directory and run the following:

pip install -e .

To check the installation, run the unit test suite:

bash run_all_tests.sh -v