From 67fc206c680de5f91d8041e1d3d27f5017cd02b8 Mon Sep 17 00:00:00 2001
From: "Peter Marek (marekp)" <marekp@coventry.ac.uk>
Date: Tue, 23 Mar 2021 10:23:55 +0000
Subject: [PATCH] add gputest script

---
 gputest.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 gputest.py

diff --git a/gputest.py b/gputest.py
new file mode 100644
index 0000000..3d80681
--- /dev/null
+++ b/gputest.py
@@ -0,0 +1,12 @@
+from distutils.version import LooseVersion
+import warnings
+import tensorflow as tf
+
+# Check tensorflow version
+print(f'Tensorflow version: {tf.__version__}')
+
+# Check for GPU
+if not tf.test.gpu_device_name():
+        warnings.warn('No GPU found')
+else:
+     	print(f'Default GPU device: {tf.test.gpu_device_name()}')