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
.