Adding check program for GPU capabilities.

This commit is contained in:
2024-04-08 12:44:46 -04:00
parent 497fe02462
commit 6e56019325

11
Models/check_gpu.py Normal file
View File

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