Tuesday, 15 September 2020

HOW TO SEE GPU AVAILABLITY

 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")

No comments:

Post a Comment