How to set specific gpu in tensorflow?

by

There are 3 ways to achieve this:

  1. Using CUDA_VISIBLE_DEVICES environment variable.
    by setting environment variable CUDA_VISIBLE_DEVICES="1" makes only device 1 visible and by setting CUDA_VISIBLE_DEVICES="0,1" makes devices 0 and 1 visible. You can do this in python by having a line os.environ["CUDA_VISIBLE_DEVICES"]="0,1" after importing os package.

  2. Using with tf.device('/gpu:2') and creating the graph. Then it will use GPU device 2 to run.

  3. Using config = tf.ConfigProto(device_count = {'GPU': 1}) and then sess = tf.Session(config=config). This will use GPU device 1.

More Related Contents:

  1. TensorFlow and person recognition in video stream
  2. How to compile Tensorflow with SSE4.2 and AVX instructions?
  3. How to understand static shape and dynamic shape in TensorFlow?
  4. Can I use TensorBoard with Google Colab?
  5. How do I split Tensorflow datasets?
  6. After building TensorFlow from source, seeing libcudart.so and libcudnn errors
  7. Convert a graph proto (pb/pbtxt) to a SavedModel for use in TensorFlow Serving or Cloud ML Engine
  8. How to create dataset in the same format as the FSNS dataset?
  9. Specify connections in NN (in keras)
  10. What are the differences between all these cross-entropy losses in Keras and TensorFlow?
  11. Tensorflow: ImportError: libcusolver.so.8.0: cannot open shared object file: No such file or directory
  12. Tensorflow crashes with CUBLAS_STATUS_ALLOC_FAILED
  13. Does model.compile() initialize all the weights and biases in Keras (tensorflow backend)?
  14. Keras Loss Function with Additional Dynamic Parameter
  15. TensorFlow: How to measure how much GPU memory each tensor takes?
  16. How to convert .pb to TFLite format?
  17. How to select rows from a 3-D Tensor in TensorFlow?
  18. Working with multiple graphs in TensorFlow
  19. Simple way to visualize a TensorFlow graph in Jupyter?
  20. ValueError: Shapes (None, 1) and (None, 2) are incompatible
  21. Update only part of the word embedding matrix in Tensorflow
  22. Could not install packages due to an EnvironmentError: [WinError 5] Access is denied:
  23. Keras split train test set when using ImageDataGenerator
  24. CUDA_ERROR_OUT_OF_MEMORY in tensorflow
  25. Implementing contrastive loss and triplet loss in Tensorflow
  26. Confusion about keras Model: __call__ vs. call vs. predict methods
  27. What does batch, repeat, and shuffle do with TensorFlow Dataset?
  28. How to replace (or insert) intermediate layer in Keras model?
  29. Choosing number of Steps per Epoch
  30. Parallelism isn’t reducing the time in dataset map

Leave a Comment