Metadata-Version: 2.4
Name: torch_bench_gpu
Version: 0.1.2
Summary: Tiny PyTorch GPU-benchmark CLI
Author-email: Arnaud Pannatier <arn.pannatier@gmail.com>
Requires-Python: >=3.8
Requires-Dist: torch>=2.8.0.dev20250515
Description-Content-Type: text/markdown

# Torch Bench GPU

Torch Bench GPU is a benchmarking tool for evaluating the performance of a GPU under pytorch. 

It helps you measure the actual GFLOPS of your GPU in different regime. 

you can run it without installing it by using the following command:

```bash
uvx --from torch-bench-gpu --extra-index-url https://download.pytorch.org/whl/nightly/cu126 torch_bench_gpu
```


# To add the right torch

```bash
uv add --pre torch --default-index https://pypi.org/simple --index https://download.pytorch.org/whl/nightly/cu126
```
