Skip to main content

icon

torchinstaller

Installing PyTorch has never been easier

PYPI Version Licence

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-geometric can 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)

Source distribution for torchinstaller 0.9.1
File Size Uploaded
torchinstaller-0.9.1.tar.gz 58.1 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for torchinstaller 0.9.1
File Interpreter ABI Platform
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

Release history Release notifications | RSS feed

This release

0.9.1 This release

2 release files

0.9.0

2 release files

0.8.9

2 release files

0.8.8

2 release files

0.8.7

2 release files

0.8.6

2 release files

0.8.5

2 release files

0.8.4

2 release files

0.8.1

2 release files

0.8.0

2 release files

0.7.1

2 release files

0.7.0

2 release files

0.6.0

2 release files

0.5.10

2 release files

0.5.8

2 release files

0.5.7

2 release files

0.5.6

2 release files

0.4.4

2 release files

0.4.1

2 release files

0.4.0

2 release files

0.3.3

2 release files

0.3.2

2 release files

0.3.1

2 release files

0.3.0

2 release files

0.2.9

2 release files

0.2.8

2 release files

0.2.7

2 release files

0.2.6

2 release files

0.2.5

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page