Installation

Requirements

Python 3.8 or higher
PyTorch 1.12 or higher
CUDA (optional, for GPU support)

Basic Installation

Install from PyPI using pip:

pip install deepml

With GPU Support

Before installing deep-ml, install PyTorch with CUDA support. Visit the PyTorch installation page to get the appropriate command for your system.

For example, with CUDA 11.8:

pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118
pip install deepml

Development Installation

To install from source for development:

git clone https://github.com/sagar100rathod/deep-ml.git
cd deep-ml
pip install -e ".[dev]"

Optional Dependencies

Distributed Training

For Lightning Fabric support:

pip install lightning-fabric

For HuggingFace Accelerate support:

pip install accelerate

Experiment Tracking

For MLflow:

pip install mlflow

For Weights & Biases:

pip install wandb

Data Augmentation

For advanced augmentations with Albumentations:

pip install albumentations

Verifying Installation

To verify your installation:

import deepml
print(deepml.__version__)

# Check available trainers
from deepml.fabric_trainer import FabricTrainer
from deepml.accelerator_trainer import AcceleratorTrainer
print("Installation successful!")