Utilities for GPU activation and data transfer in PyTorch.
Project description
get_gpu
get_gpu is a Python package that simplifies GPU management in PyTorch projects. It helps users activate a GPU (if available) and move data to GPU memory for faster training and inference.
Features
- Automatically detect and use the GPU if available, else fallback to CPU.
- Transfer tensors or datasets to the selected device.
- Wrap PyTorch DataLoader to seamlessly move data to the GPU during iteration.
Installation
Install the package directly from PyPI:
pip install get-gpu
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
gpu-active-0.1.0.tar.gz
(2.9 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file gpu-active-0.1.0.tar.gz.
File metadata
- Download URL: gpu-active-0.1.0.tar.gz
- Upload date:
- Size: 2.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.11.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
228891b0cc5ad8eccadbc5a67b7287f49c01a604bd8bff3004c60d6d87dd4bc1
|
|
| MD5 |
b8b7d6b7459ecf5fe29fabab2b4b6a35
|
|
| BLAKE2b-256 |
8ac564b0b98f0ace06418a0dc4c94e45caae5aa58c87c75b4e210fc23fa48932
|
File details
Details for the file gpu_active-0.1.0-py3-none-any.whl.
File metadata
- Download URL: gpu_active-0.1.0-py3-none-any.whl
- Upload date:
- Size: 3.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.11.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
02451e1610e6afd3ef9f234afeb5f1320f72ee4f67229be21695e54a1afc4523
|
|
| MD5 |
6c61d936560c87eb98ccd5778a802549
|
|
| BLAKE2b-256 |
7998ae3f8b58b093e492deb943cbc631792b4c3927f7bcf3f3f8302558b4f9d0
|
