6 lines
176 B
Python
6 lines
176 B
Python
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")
|