Simple utility to install pytorch, pytorch-geometric and pytorch-lightning. Detects CUDA version automatically.
Project description
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] [--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
Project details
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
torchinstaller-0.3.1.tar.gz
(4.4 kB
view hashes)
Built Distribution
Close
Hashes for torchinstaller-0.3.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 731ffe5f254efbf6878c078a2a00f5894ad1edfdb890f1dce4aa65a9227c1957 |
|
MD5 | ffe22d15b65076a00f56716404113ca0 |
|
BLAKE2b-256 | 725133d81ca98aa594070e51a5373fc16df8b4e7c60cb674365e4e2f5a0f6eb5 |