kirin-docs/setup_env.sh at master · mad1081/kirin-docs · GitHub

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
#!/bin/bash
# InnovateX Document Detection - Linux/macOS Setup Script
# Run this script to set up your development environment

set -e  # Exit on error

echo "========================================"
echo "InnovateX Document Detection Setup"
echo "========================================"
echo ""

# Check if conda is installed
if ! command -v conda &> /dev/null; then
    echo "Error: Conda not found. Please install Anaconda or Miniconda."
    echo "Download from: https://www.anaconda.com/download"
    exit 1
fi

echo "Step 1: Creating conda environment..."
conda create -n innovatex python=3.9 -y

echo ""
echo "Step 2: Activating environment..."
source "$(conda info --base)/etc/profile.d/conda.sh"
conda activate innovatex

echo ""
echo "Step 3: Installing PyTorch with CUDA 12.1..."
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

echo ""
echo "Step 4: Installing YOLOv5 and dependencies..."
pip install yolov5

echo ""
echo "Step 5: Installing additional requirements..."
pip install pdf2image pillow opencv-python pandas matplotlib jupyter

echo ""
echo "Step 6: Installing Poppler (for PDF conversion)..."
if [[ "$OSTYPE" == "linux-gnu"* ]]; then
    echo "Detected Linux. Installing poppler-utils..."
    sudo apt-get update && sudo apt-get install -y poppler-utils
elif [[ "$OSTYPE" == "darwin"* ]]; then
    echo "Detected macOS. Installing poppler via Homebrew..."
    if command -v brew &> /dev/null; then
        brew install poppler
    else
        echo "Warning: Homebrew not found. Please install Poppler manually."
        echo "Visit: https://formulae.brew.sh/formula/poppler"
    fi
else
    echo "Warning: Unknown OS. Please install Poppler manually."
fi

echo ""
echo "Step 7: Verifying installation..."
python -c "import torch; print(f'PyTorch: {torch.__version__}'); print(f'CUDA Available: {torch.cuda.is_available()}'); print(f'GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"None\"}')"

echo ""
echo "========================================"
echo "Setup Complete!"
echo "========================================"
echo ""
echo "Environment 'innovatex' is ready to use."
echo ""
echo "To activate the environment, run:"
echo "    conda activate innovatex"
echo ""
echo "To start Jupyter Notebook, run:"
echo "    jupyter notebook"
echo ""