Installation

To install the repository with Dali and/or UMAP support, use:

pip3 install .[dali,umap]

If no Dali/UMAP support is needed, the repository can be installed as:

pip3 install .

NOTE: If you want to modify the library, install it in dev mode with -e.

NOTE 2: Soon to be on pip.

Requirements:

torch
tqdm
einops
wandb
pytorch-lightning
lightning-bolts

# Optional
nvidia-dali

NOTE: if you are using CUDA 10.X use nvidia-dali-cuda100 in requirements.txt.