A tool to automatically install PyTorch packages with CUDA support if available
Project description
PyTorch CUDA Installer
PyTorch CUDA Installer is a Python package that simplifies the process of installing PyTorch packages with CUDA support. It automatically detects the available CUDA version on your system and installs the appropriate PyTorch packages.
Installation
You can install the PyTorch CUDA Installer using pip:
pip install torch-cuda-installer
Usage
After installation, you can use the package in two ways:
-
As a command-line tool:
torch-cuda-installer --torch --torchvision --torchaudio -
As a Python module:
from torch_cuda_installer import install_pytorch install_pytorch(cuda_key=None, packages=['torch', 'torchvision', 'torchaudio'])
Use any combination of --torch, --torchvision, and --torchaudio to select which packages to install.
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
Built Distribution
File details
Details for the file torch_cuda_installer-0.1.0.tar.gz.
File metadata
- Download URL: torch_cuda_installer-0.1.0.tar.gz
- Upload date:
- Size: 3.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.8.19
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | fea771c493bc79140e5b2183df267fafb0d51bd578253c40dbe76ebe1592006d |
|
| MD5 | e5cf126cd313a11395af40920228b5a2 |
|
| BLAKE2b-256 | 2c4d68523a42e3240b600b8ef1b82510cf40d5c3ade7e2106c99422f2df3f7d2 |
File details
Details for the file torch_cuda_installer-0.1.0-py3-none-any.whl.
File metadata
- Download URL: torch_cuda_installer-0.1.0-py3-none-any.whl
- Upload date:
- Size: 3.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.8.19
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 1c2312e30624b01516207140894583403a2466623d312da5deed55599623ec8c |
|
| MD5 | 69a9c3abab987d1759d0a7c0a16fb550 |
|
| BLAKE2b-256 | d6b6663168d50f9ee570f798356602d4d2eb6e014494c022371e4a04990d707e |
