tensorflow if on gpu code example
Example 1: check if tensorflow gpu is installed
import tensorflow as tf
print(tf.test.gpu_device_name())
Example 2: tensorflow check gpu
tf.config.list_physical_devices('GPU')
import tensorflow as tf
print(tf.test.gpu_device_name())
tf.config.list_physical_devices('GPU')