You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
6 lines
189 B
6 lines
189 B
2 years ago
|
import tensorflow as tf
|
||
|
|
device_name = tf.test.gpu_device_name()
|
||
|
|
if device_name != '/device:GPU:0':
|
||
|
|
raise SystemError('GPU device not found')
|
||
|
|
print('Found GPU at: {}'.format(device_name))
|