Small utils for Tensorflow in Python.
Project description
Tensorutils
Small utils for Tensorflow in Python.
Select CPU if not GPU
A function to force Tensorflow to use CPU instead of GPU if this is not present. This function is very easy to use. For example:
from tensorutils import select_cpu_if_not_gpu
# With static memory size
select_cpu_if_not_gpu()
# With dynamic memory size
select_cpu_if_not_gpu(True)
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
tensorutils-0.0.3.tar.gz
(2.2 kB
view hashes)