install_env.sh

# Run this code manually
# conda create -n faiss_env python=3.10
# conda activate faiss_env

conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.1 -c pytorch -c nvidia -y
pip install transformers datasets pyserini

conda install -c pytorch -c nvidia faiss-gpu=1.8.0 -y

pip install uvicorn fastapi huggingface_hub pydantic hf_transfer