tensorflow

TensorFlow test programs

  1. 1
  2. 2
  3. 3
  4. 4
  5. 5
import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")