Install PyTorch distributions from the latest wheels.
Project description
Disclaimer
Neither this project (pytorch_wheel_installer) nor its author (Philip Meier) are affiliated with PyTorch in any way. PyTorch and any related marks are trademarks of Facebook, Inc.
pytorch_wheel_installer
Commandline utility and tox -plugin to install PyTorch distributions from the latest wheels. The computation backend (CPU, CUDA), the language version, and the platform are detected automatically but can be overwritten manually.
package |
|
---|---|
code |
|
tests |
Installation
pytorch_wheel_installer is not yet listed on PyPI .
You can install the latest version with
pip install git+https://github.com/pmeier/pytorch_wheel_installer
Usage
CLI
The CLI can be invoked by pytorch_wheel_installer or its shorthand pwi.
$ pwi --help
usage: pwi [-h] [--version] [--distribution DISTRIBUTION] [--backend BACKEND]
[--language LANGUAGE] [--platform PLATFORM] [--no-install]
[--pip-cmd PIP_CMD]
Install PyTorch from the latest wheels.
optional arguments:
-h, --help show this help message and exit
--version, -v Show version and exit.
--distribution DISTRIBUTION, -d DISTRIBUTION
PyTorch distribution e.g. 'torch', 'torchvision'.
Multiple distributions can be given as a comma-
separated list. Defaults to 'torch,torchvision'.
--backend BACKEND, -b BACKEND
Computation backend e.g. 'cpu' or 'cu102'. If not
given the backend is automatically detected from the
available hardware preferring CUDA over CPU.
--language LANGUAGE, -l LANGUAGE
Language implementation and version tag e.g. 'py3',
'cp36'. Defaults to the language version used to run
this.
--platform PLATFORM, -p PLATFORM
Platform e.g. 'linux', 'windows', 'macos', or 'any'.
Defaults to the platform that is used to run this.
--no-install, -ni If given, the selected wheels are written to STDOUT
instead of installed.
--pip-cmd PIP_CMD, -pc PIP_CMD
pip command that is used to install the wheels.
Defaults to 'pip install'
The --no-install option can be used to pipe or redirect the wheel links such as generating a requirements.txt file:
$ pwi --no-install > requirements.txt
$ cat requirements.txt
https://download.pytorch.org/whl/cu102/torch-1.5.1-cp36-cp36m-linux_x86_64.whl
https://download.pytorch.org/whl/cu102/torchvision-0.6.1-cp36-cp36m-linux_x86_64.whl
tox
$ tox --help
...
optional arguments:
...
--pytorch-install Install PyTorch distributions from the latest
wheels before the other dependencies. (default:
False)
--pytorch-distribution DISTRIBUTION
PyTorch distribution e.g. 'torch', 'torchvision'.
Multiple distributions can be given as a
comma-separated list. Defaults to
'torch,torchvision'. (default:torch,torchvision)
--pytorch-backend BACKEND Computation backend e.g. 'cpu' or 'cu102'. If not
given the backend is automatically detected from the
available hardware preferring CUDA over CPU.
(default: None)
--pytorch-language LANGUAGE Language implementation and version tag e.g. 'py3',
'cp36'. Defaults to the language version used to run
this. (default: None)
--pytorch-platform PLATFORM Platform e.g. 'linux', 'windows', 'macos', or 'any'.
Defaults to the platform that is used to run this.
(default: None)
...
If --pytorch-install is not given, nothing is installed.
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
Built Distribution
Hashes for pytorch_wheel_installer-0.1.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | d08c13a4bbbca6d4ef14911c657a12b29f098ccc232444c5bc09cc6e0951ace9 |
|
MD5 | 6ae2ae3153db5404d7eef6791aca01a4 |
|
BLAKE2b-256 | d374a8e605505c881917a4eeab1a3168a1e91637ac8b7d4bff29861645fd55ea |
Hashes for pytorch_wheel_installer-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b24bdf612f2267b5957d89eae79c83ad372e75370e1c1650e3454f92f4e006ba |
|
MD5 | 615f6c036bf0305ae0a29dcd474036cd |
|
BLAKE2b-256 | 316608b42104ad06bc37b1a7f04fcc39830037519e05a3eff873d84fd7dad30e |