Automatically detects your GPU and installs the correct version of PyTorch for your environment
Project description
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file gaff_gpu-0.1.0.tar.gz.
File metadata
- Download URL: gaff_gpu-0.1.0.tar.gz
- Upload date:
- Size: 7.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1e5d0af05ae47d03f32e6f0e3f1485bb6e8fa91d1f57d7448f6a733a6df0d7db
|
|
| MD5 |
07b78793d01df7ea5518539d7c1de02d
|
|
| BLAKE2b-256 |
67497c1cd111f884269c67a88324e88800ed17abdae1604e1ffedc23c60a0191
|
File details
Details for the file gaff_gpu-0.1.0-py3-none-any.whl.
File metadata
- Download URL: gaff_gpu-0.1.0-py3-none-any.whl
- Upload date:
- Size: 8.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
a88a3c826425f698a7f430d7df7dee0eac20f61cb8479a6a34ab710c32016966
|
|
| MD5 |
87df5b878774a783a15c42587b31dac0
|
|
| BLAKE2b-256 |
c81e8a41ec294037a7d704c1d893f81b98922b2083c8b888f073ae69f5154e85
|
