File size: 1,138 Bytes
9ebdc51 5f0ab61 9ebdc51 5f0ab61 9ebdc51 5f0ab61 9ebdc51 |
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 33 34 35 36 37 38 39 |
#!/bin/bash
# OpenTrack build-time setup script
# This script clones OpenTrack and installs its dependencies at Docker build time
set -e # Exit on error
echo "π€ Setting up OpenTrack at build time..."
echo "=========================================="
# Clone OpenTrack repository
OPENTRACK_DIR="$HOME/OpenTrack"
if [ ! -d "$OPENTRACK_DIR" ]; then
echo "π¦ Cloning OpenTrack repository..."
git clone https://github.com/GalaxyGeneralRobotics/OpenTrack.git "$OPENTRACK_DIR"
echo "β Repository cloned to $OPENTRACK_DIR"
else
echo "β OpenTrack repository already exists"
fi
# Install OpenTrack-specific requirements (most are already in main requirements.txt)
# Main requirements.txt already has: jax[cuda12], mujoco, brax, playground, mediapy
echo ""
echo "π Installing additional OpenTrack dependencies..."
pip install --no-cache-dir \
tyro \
opencv-python \
wandb \
jaxopt \
flax \
absl-py \
joblib \
imageio[ffmpeg]
echo "β OpenTrack dependencies installed"
echo ""
echo "=========================================="
echo "β
OpenTrack build-time setup complete!"
echo ""
|