mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-03-09 10:12:08 -05:00
3215159726
Created unified setup-environment.sh script that: - Detects Apple Silicon and creates arm64-optimized venv - Handles all dependencies automatically - Creates activation helper with architecture awareness - Works across macOS (Intel/Apple Silicon), Linux, Windows Updated all documentation to use ONE setup command: - README.md: Updated Quick Start - docs/STUDENT_QUICKSTART.md: Updated Getting Started - book/quickstart-guide.md: Updated 2-Minute Setup Enhanced tito setup command with: - Apple Silicon detection (checks for Rosetta vs native) - Automatic arm64 enforcement when on Apple Silicon - Architecture verification after venv creation - Changed venv path from tinytorch-env to standard .venv Students now have ONE clear path: ./setup-environment.sh
99 lines
2.6 KiB
Bash
Executable File
99 lines
2.6 KiB
Bash
Executable File
#!/bin/bash
|
|
# TinyTorch Environment Setup
|
|
# Single canonical way to set up TinyTorch for development
|
|
|
|
set -e # Exit on error
|
|
|
|
echo "🔥 TinyTorch Environment Setup"
|
|
echo "================================"
|
|
echo ""
|
|
|
|
# Detect system
|
|
OS=$(uname -s)
|
|
ARCH=$(uname -m)
|
|
|
|
echo "📋 System Info:"
|
|
echo " OS: $OS"
|
|
echo " Architecture: $ARCH"
|
|
echo ""
|
|
|
|
# Check if on Apple Silicon
|
|
if [ "$OS" = "Darwin" ] && [ "$ARCH" = "arm64" ]; then
|
|
echo "✅ Detected Apple Silicon (arm64)"
|
|
PYTHON_CMD="arch -arm64 /usr/bin/python3"
|
|
elif [ "$OS" = "Darwin" ]; then
|
|
echo "⚠️ On macOS but not arm64 - using system Python"
|
|
PYTHON_CMD="/usr/bin/python3"
|
|
else
|
|
echo "📦 Using system Python"
|
|
PYTHON_CMD="python3"
|
|
fi
|
|
|
|
# Create venv
|
|
echo ""
|
|
echo "🐍 Creating virtual environment..."
|
|
if [ -d ".venv" ]; then
|
|
echo "⚠️ .venv already exists - removing it"
|
|
rm -rf .venv
|
|
fi
|
|
|
|
$PYTHON_CMD -m venv .venv
|
|
echo "✅ Virtual environment created"
|
|
|
|
# Activate and install
|
|
echo ""
|
|
echo "📦 Installing dependencies..."
|
|
|
|
if [ "$OS" = "Darwin" ] && [ "$ARCH" = "arm64" ]; then
|
|
# On Apple Silicon, use arch prefix for all pip commands
|
|
arch -arm64 .venv/bin/pip install --upgrade pip -q
|
|
arch -arm64 .venv/bin/pip install -r requirements.txt -q
|
|
arch -arm64 .venv/bin/pip install -e . -q
|
|
else
|
|
.venv/bin/pip install --upgrade pip -q
|
|
.venv/bin/pip install -r requirements.txt -q
|
|
.venv/bin/pip install -e . -q
|
|
fi
|
|
|
|
echo "✅ Dependencies installed"
|
|
|
|
# Verify
|
|
echo ""
|
|
echo "🔍 Verifying installation..."
|
|
if [ "$OS" = "Darwin" ] && [ "$ARCH" = "arm64" ]; then
|
|
VENV_ARCH=$(arch -arm64 .venv/bin/python3 -c "import platform; print(platform.machine())")
|
|
else
|
|
VENV_ARCH=$(.venv/bin/python3 -c "import platform; print(platform.machine())")
|
|
fi
|
|
echo " Python architecture: $VENV_ARCH"
|
|
|
|
# Create activation helper
|
|
cat > activate.sh << 'EOF'
|
|
#!/bin/bash
|
|
# TinyTorch activation helper
|
|
if [ "$(uname -s)" = "Darwin" ] && [ "$(uname -m)" = "arm64" ]; then
|
|
# On Apple Silicon, ensure arm64
|
|
export TINYTORCH_ARCH="arm64"
|
|
alias python='arch -arm64 .venv/bin/python3'
|
|
alias pip='arch -arm64 .venv/bin/pip'
|
|
source .venv/bin/activate
|
|
echo "🔥 TinyTorch environment activated (arm64)"
|
|
else
|
|
source .venv/bin/activate
|
|
echo "🔥 TinyTorch environment activated"
|
|
fi
|
|
echo "💡 Try: tito system doctor"
|
|
EOF
|
|
|
|
chmod +x activate.sh
|
|
|
|
echo ""
|
|
echo "✅ Setup complete!"
|
|
echo ""
|
|
echo "🚀 Next steps:"
|
|
echo " 1. source activate.sh # Activate environment"
|
|
echo " 2. tito system doctor # Verify setup"
|
|
echo " 3. tito module view 01_tensor # Start learning"
|
|
echo ""
|
|
|