Metadata-Version: 2.1
Name: torchinstaller
Version: 0.4.1
Summary: Simple utility to install pytorch, pytorch-geometric and pytorch-lightning. Detects CUDA version automatically.
Project-URL: repository, https://github.com/dk0d/torchinstaller.git
Author-email: Daniel Capecci <7775585+dk0d@users.noreply.github.com>
License: MIT
Requires-Python: <4.0,>=3.9
Requires-Dist: tomlkit<1.0.0,>=0.11.6
Description-Content-Type: text/markdown

# PyTorch Install Helper

Super simple helper to install pytorch stuff without having to check cuda versions and go to websites for the installer URLs.
Installs based on requested or detected CUDA version, doesn't check python versions.

> _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] [--pytorch] [--pyg] [--pyg-lib-source] [--cuda {10.2,11.3,11.6,11.7}] [--lightning]
                    [--use {pip,conda,mamba,poetry}] [-install]

options:
  -h, --help            show this help message and exit
  --pytorch, -pt        Flag to install pytorch
  --pyg, -pyg           Flag to install pytorch-geometric
  --pyg-lib-source, -pyg-src
                        Pytorch Geometric doesn't support wheels for M1/M2 macs, they recommend installing
                        from source
  --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 install pytorch-lightning
  --use {pip,conda,mamba,poetry}, -use {pip,conda,mamba,poetry}
                        set command to install with
  -install, -i          Run installation (default is to dry run commands)
```

> Note: pytorch-geometric can be problematic to install.  Install from source has been added to facilitate installation, but referring to their documentation may be the only way to address any errors in installation if they occur during installation.

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

