File size: 826 Bytes
b23769d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | # Setup Instructions
## Set Up Conda Environment
```bash
# Create and activate conda environment
conda create -n capvector-openvla-oft python=3.10 -y
conda activate capvector-openvla-oft
# Install PyTorch
# Use a command specific to your machine: https://pytorch.org/get-started/locally/
pip3 install torch torchvision torchaudio
# Clone openvla-oft repo and pip install to download dependencies
git clone https://github.com/Songwxuan/CapVector
cd openvla-oft
pip install -e .
# Install Flash Attention 2 for training (https://github.com/Dao-AILab/flash-attention)
# =>> If you run into difficulty, try `pip cache remove flash_attn` first
pip install packaging ninja
ninja --version; echo $? # Verify Ninja --> should return exit code "0"
pip install "flash-attn==2.5.5" --no-build-isolation
``` |