torchinstaller is a super simple helper to install PyTorch stuff without having to check cuda versions and go to websites for the installer URLs.
It installs PyTorch components based on requested or detected CUDA version, and doesn't check python versions.
Only Linux and macOS supported
It 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 --help
Usage: torchinstall [OPTIONS] COMMAND [ARGS]...
╭─ Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --install-completion Install completion for the current shell. │
│ --show-completion Show completion for the current shell, to copy it or customize the installation. │
│ --help Show this message and exit. │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Commands ─────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ platforms List available compute platforms │
│ install Install PyTorch and related libraries │
│ sync Update installation commands by parsing the pytorch website │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
List the platforms that have been synced via torchinstall sync
$ torchinstall platforms
Available Compute Platforms:
- cu102
- cu111
- cu113
- cu116
- cu117
- cu118
- cu121
- cu124
- cu126
- cu128
- rocm4.0.1
- rocm4.1
- rocm4.2
- rocm4.5.2
- rocm5.1.1
- rocm5.2
- rocm5.4.2
- rocm5.6
- rocm5.7
- rocm6.0
- rocm6.1
- rocm6.2
- rocm6.2.4
- rocm6.3
Note:
pytorch-geometriccan be problematic to install. Installing from source has been added to facilitate installation, but referring to their documentation may be necessary to address errors if they occur.
Release files for torchinstaller 0.9.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| torchinstaller-0.9.1.tar.gz | 58.1 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| torchinstaller-0.9.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size:85.9 kB
Release files / torchinstaller-0.9.1.tar.gz
| Download URL | torchinstaller-0.9.1.tar.gz |
|---|---|
| Size | 58.1 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
a6c1c017e2cd6f0a18fd3c91b35e0e5304a2592915e6eac8fe59cd854d182061
|
|
BLAKE2b-256 checksum How to use checksums |
2a85bb5718f3edc17dc594286f86f6f08dd523269ec2de12f18b36916fa8eaed
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.8.0
|
Release files / torchinstaller-0.9.1-py3-none-any.whl
| Download URL | torchinstaller-0.9.1-py3-none-any.whl |
|---|---|
| Size | 27.8 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
14ce7d99e19926d76f0b40ae2f836ca2fe038a3171e1b9deebaef6f09675127e
|
|
BLAKE2b-256 checksum How to use checksums |
83dacef4f4993d5bb2a88aac705336c589b6c09465f79272b4bade7c7ba07afc
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.8.0
|