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 ""