File size: 410 Bytes
48ee475 | 1 2 3 4 5 6 7 8 9 10 11 12 | #!/usr/bin/env bash
# One-time setup: creates a venv and installs dependencies (CUDA 12.4 wheels).
set -e
cd "$(dirname "$0")"
python3 -m venv venv
source venv/bin/activate
python -m pip install --upgrade pip
pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
pip install -r requirements.txt
echo
echo "Done. Run: ./start_linux.sh (add --nf4 for ~7GB VRAM mode)"
|