torchinstaller 0.2.8

Simple utility to install pytorch, pytorch-geometric and pytorch-lightning. Detects CUDA version automatically.

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

pip install torchinstaller

Usage

$ torchinstall -h
usage: torchinstall [-h] [--poetry] [--dryrun] [--pyg] [--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
  --lightning, -l  Flag to also install pytorch-lightning

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