Tensorflow set CUDA_VISIBLE_DEVICES within jupyter

by

You can set environment variables in the notebook using os.environ. Do the following before initializing TensorFlow to limit TensorFlow to first GPU.

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"

You can double check that you have the correct devices visible to TF

from tensorflow.python.client import device_lib
print device_lib.list_local_devices()

I tend to use it from utility module like notebook_util

import notebook_util
notebook_util.pick_gpu_lowest_memory()
import tensorflow as tf

More Related Contents:

  1. Make a custom loss function in keras
  2. Could not load dynamic library ‘cudart64_101.dll’ on tensorflow CPU-only installation
  3. Disable assertions in Python
  4. Keras replacing input layer
  5. Reproducible results in Tensorflow with tf.set_random_seed
  6. Keras Sequential model with multiple inputs
  7. nltk doesn’t add $NLTK_DATA to search path?
  8. This model has not yet been built error on model.summary()
  9. How can I use a pre-trained neural network with grayscale images?
  10. How to understand the term `tensor` in TensorFlow?
  11. Tensorflow r1.0 : could not a find a version that satisfies the requirement tensorflow
  12. TensorFlow ValueError: Cannot feed value of shape (64, 64, 3) for Tensor u’Placeholder:0′, which has shape ‘(?, 64, 64, 3)’
  13. TypeError: ‘Tensor’ object does not support item assignment in TensorFlow
  14. TensorFlow – numpy-like tensor indexing
  15. How do I set TensorFlow RNN state when state_is_tuple=True?
  16. Set LD_LIBRARY_PATH before importing in python
  17. In Tensorflow, get the names of all the Tensors in a graph
  18. Custom loss function in Keras
  19. Save and load model optimizer state
  20. Keras misinterprets training data shape
  21. In TensorFlow, what is the difference between Session.run() and Tensor.eval()?
  22. Why can I not import Tensorflow.contrib I get an error of No module named ‘tensorflow.python.saved
  23. How do I get the weights of a layer in Keras?
  24. Environment Variables when script run by cron
  25. Installing TensorFlow on Windows (Python 3.6.x)
  26. How to fix “AttributeError: module ‘tensorflow’ has no attribute ‘get_default_graph'”?
  27. Loading environment modules within a python script
  28. What is right batch normalization function in Tensorflow?
  29. Strange behaviour of the loss function in keras model, with pretrained convolutional base
  30. Bundling Data files with PyInstaller 2.1 and MEIPASS error –onefile

Leave a Comment