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') +