Installation

Requirements

• Python 3.10+

• PyTorch 2.0+

• CUDA 11.8+ (optional, for GPU acceleration)

From PyPI

pip install erasus

From Source

git clone https://github.com/onepunchmonk/erasus.git
cd erasus
pip install -e .

GPU Support

# Install with CUDA 12.1
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install erasus

Development Installation

git clone https://github.com/onepunchmonk/erasus.git
cd erasus
pip install -e ".[dev]"

# Or use the setup script:
bash scripts/setup_env.sh       # CPU only
bash scripts/setup_env.sh --gpu # With CUDA

Docker

# Build and run tests
docker compose -f docker/docker-compose.yml up test

# Development shell
docker compose -f docker/docker-compose.yml run dev

# GPU benchmarks
docker compose -f docker/docker-compose.yml up benchmark

Verifying Installation

import erasus
print("Erasus OK")

import torch
print(f"PyTorch: {torch.__version__}")
print(f"CUDA: {torch.cuda.is_available()}")