Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Add the ability to specify gpu_options

See original GitHub issue

TensorFlow.js version

0.12.6

Browser version

Node.js 10.9

Describe the problem or feature request

I am using tfjs-node-gpu and it allocates the whole memory of my GPU (the default TF behavior). I request to have an ability to specify the backend options, particularly gpu_options.allow_growth.

Issue Analytics

State:
Created 5 years ago
Reactions:2
Comments:16 (2 by maintainers)

Top GitHub Comments

5reactions

cjuecommented, May 22, 2019

There appears to be a workaround for setting allow_growth through an environment variable. This works for my tfjs-node-gpu project:

#Limit GPU memory usage with an environment variable
export TF_FORCE_GPU_ALLOW_GROWTH=true

4reactions

iacovlev-pavelcommented, Jan 17, 2019

+1, having access to gpu_options would be great.

Top Results From Across the Web

How to use tensorflow-gpu GPUOptions - Stack Overflow

I'm assuming that you are running tensorflow version 2.0. In which case you can set GPU options in eager execution mode like so:...

tf.compat.v1.GPUOptions | TensorFlow v2.11.0

v1. GPUOptions. Stay organized with collections Save and categorize content based on your preferences.

TensorFlow CPUs and GPUs Configuration | by Li Yin - Medium

In ConfigProto() , gpu_options is used to configure gpus. Visible Devices. First, you need to check your machine's available GPUs with nvidia-smi command....

SingularityCE Configuration Files - Documentation & Examples

Most of the configuration options are set using the file singularity.conf that defines the global configuration for SingularityCE across the entire system.

Singularity Configuration Files - Apptainer

Most of the configuration options are set using the file singularity.conf that defines the ... sudo singularity capability add --user pinger CAP_NET_RAW.