| |
| """环境自检: GPU/CUDA/pyiqa/dwt/官方import 冒烟(不加载大权重)""" |
| import os, sys |
| sys.path.insert(0, "official"); sys.path.insert(0, "src") |
| import torch |
| print("torch", torch.__version__, "cuda", torch.cuda.is_available()) |
| if torch.cuda.is_available(): |
| print("gpu", torch.cuda.get_device_name(0), "mem_GB", round(torch.cuda.get_device_properties(0).total_memory/1e9,1)) |
| try: |
| import pyiqa; print("pyiqa ok") |
| except Exception as e: print("pyiqa FAIL", e) |
| try: |
| import pywt; print("pywt ok") |
| except Exception as e: print("pywt missing(可选)", e) |
| try: |
| import diffusers, transformers, peft, omegaconf |
| print("diffusers/transformers/peft ok") |
| except Exception as e: print("libs FAIL", e) |
| |
| try: |
| import model, dataset, utils, forward |
| print("official imports ok") |
| except Exception as e: |
| print("official imports FAIL", e) |
| print("env_check done")
|
|
|
