conda install matplotlib joblib scikit-image scikit-learn scipy pandas ipykernel pillow termcolor
conda install umap-learn conda-forge

conda install pytorch torchvision torchaudio torchtext faiss-gpu cudatoolkit=10.2 -c pytorch

pip install wandb sentence-transformers transformers x-transformers timm click ftfy regex tqdm click pretrainedmodels
pip install git+https://github.com/openai/CLIP.git