12 lines
338 B
Python
12 lines
338 B
Python
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')
|
|
|