Adding check program for GPU capabilities.
This commit is contained in:
11
Models/check_gpu.py
Normal file
11
Models/check_gpu.py
Normal 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')
|
||||
|
||||
Reference in New Issue
Block a user