install_env.sh

# Run this code manually
conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main
conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r

# conda install python=3.10 -y
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 accelerate gradio