TensorFlow snippets

tf_snippets.md

GPUs

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1' # ignore GPUs

CPUs

• https://gist.github.com/elliottslaughter/750a27c832782f4daec8686281027de8

Multiprocessing

• https://software.intel.com/en-us/articles/recommended-settings-for-calling-intel-mkl-routines-from-multi-threaded-applications

import psutil
import os
os.environ['OMP_NUM_THREADS'] = str(psutil.cpu_count(logical=False))

import tensorflow as tf
config = tf.ConfigProto(
  inter_op_parallelism_threads=1,
  intra_op_parallelism_threads=1,
)

with tf.Session(config=config) as sess:
  pass

Logging

• https://www.tensorflow.org/api_docs/python/tf/compat/v1/logging
• https://stackoverflow.com/questions/35911252/disable-tensorflow-debugging-information

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

import tensorflow as tf
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)