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