a package for allocating tensorflow gpu memory use in python
Project description
Gpu allocator
install
$ pip install configure_gpu
This is a package for allocating gpu memory to prevent clogging the gpu memory.
Usage
# tensorflow import
import tensorflow as tf
# gpu allocator package
from configure_gpu import gpu_allocator
# configuring gpu
gpu_memory_allocator(tf, percentage_allocation=0.95)
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 Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
Close
Hashes for configure_gpu-0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 91098e48d57eae20a9e94ae058cd0909fd8f1b8ac1f396fd3a829a3dddbdd3c0 |
|
MD5 | 08500115da93a009583ee954540220c7 |
|
BLAKE2b-256 | 553b9588a0dbc76b8010a4a4488103aa6b1e81cffed415f77708f20f55e53753 |