INSTALL.md

TARA Dependencies

1. Create/activate the conda env (skip if you already have it):

   conda create -n tara python=3.10 -y
   conda activate tara
   

2. Install CUDA 12.1 PyTorch wheels (adjust the index URL if you need a different CUDA/CPU build):

   pip install --index-url https://download.pytorch.org/whl/cu121 \
     torch==2.5.1+cu121 torchvision==0.20.1+cu121 torchaudio==2.5.1+cu121
   

3. Install the remaining model dependencies:

   pip install -r requirements.txt
   

4. (Optional) Verify the install:

   python -c "import torch, transformers; print(torch.cuda.is_available(), transformers.__version__)"