From f6e9c1ce5ccf50414388268fa923beb19dab1393 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Sat, 10 Feb 2024 02:23:50 -0500 Subject: [PATCH] add debugging to sd python script --- .../Storyteller-Studio-Filter/main.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/image_generation/Storyteller-Studio-Filter/main.py b/image_generation/Storyteller-Studio-Filter/main.py index 930dced..6d7ea03 100644 --- a/image_generation/Storyteller-Studio-Filter/main.py +++ b/image_generation/Storyteller-Studio-Filter/main.py @@ -2,12 +2,27 @@ import argparse import os import pathlib import torch +import sys # Note please use type hints. from stablediffusionfe import StableDiffusionFE import random import string +def print_gpu_info(): + print('========================================') + print('Python Interpreter', sys.executable) + print('Python Version', sys.version) + print('Python Version Info', sys.version_info) + print('PyTorch Version', torch.__version__) + print('CUDA Available?', torch.cuda.is_available()) + print('CUDA Version', torch.version.cuda) + print('CUDA Device Count', torch.cuda.device_count()) + print('CUDA architectures library was compiled for', torch.cuda.get_arch_list()) + print('========================================', flush=True) + +print_gpu_info() + def generate_random_string(length=6): # Choose from uppercase, lowercase letters and digits characters = string.ascii_letters + string.digits