bpiyush commited on
Commit
3a7d6fb
·
verified ·
1 Parent(s): 8aee191

Upload INSTALL.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. INSTALL.md +21 -0
INSTALL.md ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # TARA Dependencies
2
+
3
+ 1. Create/activate the conda env (skip if you already have it):
4
+ ```bash
5
+ conda create -n tara python=3.10 -y
6
+ conda activate tara
7
+ ```
8
+ 2. Install CUDA 12.1 PyTorch wheels (adjust the index URL if you need a different CUDA/CPU build):
9
+ ```bash
10
+ pip install --index-url https://download.pytorch.org/whl/cu121 \
11
+ torch==2.5.1+cu121 torchvision==0.20.1+cu121 torchaudio==2.5.1+cu121
12
+ ```
13
+ 3. Install the remaining model dependencies:
14
+ ```bash
15
+ pip install -r requirements.txt
16
+ ```
17
+ 4. (Optional) Verify the install:
18
+ ```bash
19
+ python -c "import torch, transformers; print(torch.cuda.is_available(), transformers.__version__)"
20
+ ```
21
+