Automatically assign available hardware on the fly, in-line with PyTorch code.
Project description
autodevice
Automatically assign devices in-line with pytorch code
Usage
from autodevice import AutoDevice
x = torch.randn([200, 50]).to(AutoDevice())
Installation
pip install auto-device
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
autodevice-0.0.1.tar.gz
(2.8 kB
view hashes)
Built Distribution
Close
Hashes for autodevice-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 38f584e7b8a5d0b4ad41ecc5ece473a8a94abb64b7dfcf18411be9dc206eb12e |
|
MD5 | a77df30d7d30981bbe309a43b3d37abd |
|
BLAKE2b-256 | 0f0961503d061686352987f34d0df1ed1d03991b7c411130c0041e29397ddc14 |