heartonbit/tf_check_gpu

Created November 14, 2019 20:51

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/heartonbit/2e760942a297cc43dc3757e56d6376ed.js"></script>
Save heartonbit/2e760942a297cc43dc3757e56d6376ed to your computer and use it in GitHub Desktop.

Download ZIP

Revisions

heartonbit created this gist Nov 14, 2019.

tf_check_gpu

-Original file line number
+Diff line change
@@ -0,0 +1,5 @@
+    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))
+    print("GPU Available: ", tf.test.is_gpu_available())