Gaff

Automatically detects your GPU and installs the correct version of PyTorch for your environment.

No more manually looking up CUDA versions or copying wheel URLs from the PyTorch website.

Installation

pip install gaff-gpu

Usage

from gaff import cuda_setup

cuda_setup()

That's it. Gaff will detect your GPU, find the right PyTorch wheel, and install it.

Options

See what would be installed without actually installing

cuda_setup(dry_run=True)

Install specific packages only

cuda_setup(packages=["torch"])

What it does

• Detects NVIDIA (CUDA) and AMD (ROCm) GPUs automatically
• Falls back to CPU-only PyTorch if no GPU is found
• Installs torch, torchvision, and torchaudio by default
• Verifies the installation worked when complete

License

MIT