Metadata-Version: 2.1
Name: torchinstaller
Version: 0.3.1
Summary: Simple utility to install pytorch, pytorch-geometric and pytorch-lightning. Detects CUDA version automatically.
Home-page: https://github.com/dk0d/torchinstaller.git
Author: Daniel Capecci
Requires-Python: >=3.9,<4.0
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Dist: tomlkit (>=0.11.6,<0.12.0)
Project-URL: Repository, https://github.com/dk0d/torchinstaller.git
Description-Content-Type: text/markdown

# PyTorch Install Helper

Super basic helper to install pytorch stuff without having to check cuda versions and go to websites for the installer URLs.

> _Only Linux and macOS Supported_

Detects what cuda version is available and runs the pip command to install latest pytorch and compatible cuda version

## Installation

```bash
pip install torchinstaller
```

## Usage

```bash
$ torchinstall -h
usage: torchinstall [-h] [--poetry] [--dryrun] [--pyg] [--cuda {10.2,11.3,11.6,11.7}] [--lightning]

optional arguments:
  -h, --help            show this help message and exit
  --poetry, -p          Uses poetry for install. Creates a torch source and adds torch to pyproject.toml
  --dryrun, -d          just prints the commands that would be run
  --pyg, -pyg           Flag to also install pytorch-geometric
  --cuda {10.2,11.3,11.6,11.7}, -c {10.2,11.3,11.6,11.7}
                        Manually specify cuda version instead of auto-detect (useful for cluster installations).
  --lightning, -l       Flag to also install pytorch-lightning
```

> Note: poetry support is a work in-progress and is unstable

