From 6e560193252752e51ccba013705ce7b3568fc8f7 Mon Sep 17 00:00:00 2001
From: Sean <skessler1964@gmail.com>
Date: Mon, 8 Apr 2024 12:44:46 -0400
Subject: [PATCH] Adding check program for GPU capabilities.

---
 Models/check_gpu.py | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 Models/check_gpu.py

diff --git a/Models/check_gpu.py b/Models/check_gpu.py
new file mode 100644
index 0000000..742f339
--- /dev/null
+++ b/Models/check_gpu.py
@@ -0,0 +1,11 @@
+import sys
+import os
+import glob
+import socket
+import tensorflow as tf
+
+# This program should print [PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')] is all is well and good
+# I have left instructions for installing the CUDA stuff in the model_host.py module.
+print(tf.config.list_physical_devices('GPU'))
+print('Done')
+