File size: 3,812 Bytes
af83d87 | 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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | #!/bin/bash
# install_deps.sh — One-time install of GR00T deps on dGPU systems (x86_64 or aarch64 GB200, CUDA 12.8+)
# Requires an NVIDIA discrete GPU with a CUDA 12.x or 13.x driver already installed.
# After install, activate with: source .venv/bin/activate
set -euo pipefail
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
REPO_ROOT="$(cd "$SCRIPT_DIR/../../.." && pwd)"
# Use sudo only when not already root
SUDO=""
if [ "$(id -u)" -ne 0 ]; then
SUDO="sudo"
fi
ARCH=$(uname -m)
# ──────────────────────────────────────────────────────────────────────────────
# System dependencies
# ──────────────────────────────────────────────────────────────────────────────
# FFmpeg runtime libs — required by torchcodec at runtime
# libaio-dev — required by deepspeed async I/O ops
echo "Installing system dependencies..."
$SUDO apt-get update -qq
$SUDO apt-get install -y --no-install-recommends ffmpeg libaio-dev
# CUDA toolkit — required by deepspeed (needs CUDA_HOME / nvcc to check op compatibility)
# Skip if already installed
if [ ! -d "/usr/local/cuda" ]; then
echo "CUDA toolkit not found. Installing cuda-toolkit-12-8..."
# Add NVIDIA CUDA apt repo if not already configured
if ! apt-cache show cuda-toolkit-12-8 &>/dev/null; then
UBUNTU_VERSION=$(. /etc/os-release && echo "${VERSION_ID//.}")
# aarch64 GB200 uses the sbsa (server base system architecture) repo
if [ "$ARCH" = "aarch64" ]; then
CUDA_REPO_ARCH="sbsa"
else
CUDA_REPO_ARCH="x86_64"
fi
KEYRING_URL="https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/${CUDA_REPO_ARCH}/cuda-keyring_1.1-1_all.deb"
echo "Adding NVIDIA CUDA apt repository..."
curl -fsSL "$KEYRING_URL" -o /tmp/cuda-keyring.deb
$SUDO dpkg -i /tmp/cuda-keyring.deb
rm /tmp/cuda-keyring.deb
$SUDO apt-get update -qq
fi
$SUDO apt-get install -y --no-install-recommends cuda-toolkit-12-8
else
echo "CUDA toolkit already installed at /usr/local/cuda."
fi
# ──────────────────────────────────────────────────────────────────────────────
# Environment
# ──────────────────────────────────────────────────────────────────────────────
# Install uv if not present
if ! command -v uv &> /dev/null; then
echo "Installing uv..."
curl -LsSf https://astral.sh/uv/install.sh | sh
export PATH="$HOME/.local/bin:$PATH"
fi
# ──────────────────────────────────────────────────────────────────────────────
# Python environment
# ──────────────────────────────────────────────────────────────────────────────
cd "$REPO_ROOT"
echo "Running uv sync (torch==2.7.1+cu128 from pytorch-cu128 index)..."
uv sync
echo "Installing package in editable mode..."
uv pip install -e .
echo ""
echo "Install complete! Activate with:"
echo " source .venv/bin/activate"
|