File size: 1,076 Bytes
d0f5e0c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | #!/usr/bin/env bash
set -euo pipefail
BASE="${BASE:-/data/home/user-test/SHC/COVT}"
TRAIN_DIR="$BASE/upstream/CoVT/train"
LOG_DIR="$BASE/logs"
mkdir -p "$LOG_DIR"
cd "$TRAIN_DIR"
echo "Installing CoVT training dependencies into the user Python scope only."
echo "No system packages, drivers, CUDA, or hardware settings are changed."
echo "Log: $LOG_DIR/prepare_user_env.log"
{
date
echo "python=$(command -v python3)"
python3 --version
python3 -m pip install --user --upgrade pip
python3 -m pip install --user -r requirements.txt
python3 -m pip install --user git+https://github.com/facebookresearch/segment-anything.git
python3 -m pip install --user -r src/anchors/DepthAnything/requirements.txt
python3 -m pip install --user "deepspeed>=0.15"
python3 -m pip install --user "xformers==0.0.29.post1"
python3 - <<'PY'
mods = ["torch", "transformers", "peft", "deepspeed", "safetensors", "qwen_vl_utils"]
for name in mods:
mod = __import__(name)
print(f"{name}: {getattr(mod, '__version__', '')}")
PY
} 2>&1 | tee "$LOG_DIR/prepare_user_env.log"
|