Test: Python Tensorflow GPU

A test code to check whether tensorflow recognizes GPU
avatar
unknown
python
3 years ago
175 B
20
Indexable
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")
Editor is loading...