python gpu_info.py

import sys
import os
import subprocess
import torch
from datetime import datetime

def get_gpu_info():
    print("=" * 50)
    print("GPU 診斷報(bào)告")
    print("=" * 50)
    print(f"診斷時間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n")

    # 系統(tǒng)信息
    print("系統(tǒng)信息:")
    print(f"Python 版本: {sys.version}")
    print(f"PyTorch 版本: {torch.__version__}")
    print(f"CUDA 版本 (PyTorch): {torch.version.cuda}")
    print(f"cuDNN 版本: {torch.backends.cudnn.version()}\n")

    # CUDA 環(huán)境檢查
    print("CUDA 環(huán)境變量:")
    cuda_vars = {
        'CUDA_HOME': os.environ.get('CUDA_HOME', '未設(shè)置'),
        'CUDA_PATH': os.environ.get('CUDA_PATH', '未設(shè)置'),
        'CUDA_VISIBLE_DEVICES': os.environ.get('CUDA_VISIBLE_DEVICES', '未設(shè)置')
    }
    for var, value in cuda_vars.items():
        print(f"{var}: {value}")
    print()

    # NVIDIA-SMI 輸出
    print("NVIDIA-SMI 信息:")
    try:
        encodings = ['gbk', 'utf-8', 'iso-8859-1']
        nvidia_smi = None
        for encoding in encodings:
            try:
                nvidia_smi = subprocess.check_output(["nvidia-smi"]).decode(encoding)
                break
            except UnicodeDecodeError:
                continue
        
        if nvidia_smi:
            print(nvidia_smi)
        else:
            print("無法解碼 nvidia-smi 輸出")
    except Exception as e:
        print(f"執(zhí)行 nvidia-smi 失敗: {e}\n")

    # PyTorch GPU 信息
    print("\nPyTorch GPU 詳細(xì)信息:")
    if torch.cuda.is_available():
        print(f"檢測到 {torch.cuda.device_count()} 個 GPU 設(shè)備")
        
        for i in range(torch.cuda.device_count()):
            props = torch.cuda.get_device_properties(i)
            print(f"\nGPU {i}: {props.name}")
            print(f"├─ 計(jì)算能力: {props.major}.{props.minor}")
            print(f"├─ 總顯存: {props.total_memory / (1024**2):.1f} MB")
            print(f"├─ 多處理器數(shù)量: {props.multi_processor_count}")
            print(f"├─ 最大線程數(shù)/塊: {props.max_threads_per_multi_processor}")
            
            # 顯存使用情況
            try:
                memory_allocated = torch.cuda.memory_allocated(i) / (1024**2)
                memory_reserved = torch.cuda.memory_reserved(i) / (1024**2)
                memory_free = (props.total_memory / (1024**2)) - memory_allocated
                
                print(f"├─ 已分配顯存: {memory_allocated:.1f} MB")
                print(f"├─ 已預(yù)留顯存: {memory_reserved:.1f} MB")
                print(f"└─ 可用顯存: {memory_free:.1f} MB")
            except Exception as e:
                print(f"└─ 無法獲取顯存使用情況: {e}")
    else:
        print("未檢測到可用的 GPU 設(shè)備")
        print("\n可能的原因:")
        print("1. CUDA 驅(qū)動未正確安裝")
        print("2. PyTorch 未編譯 CUDA 支持")
        print("3. GPU 被其他進(jìn)程占用")
        print("4. 系統(tǒng)未正確識別 GPU")

def get_usb_controller_info():
    print("\nUSB 控制器信息:")
    try:
        result = subprocess.check_output(["wmic", "path", "Win32_USBController", "get", "name,manufacturer"], 
                                      encoding='gbk')
        controllers = result.strip().split('\n')[1:]  # Skip header
        nvidia_controllers = []
        
        for controller in controllers:
            if controller.strip():
                if "NVIDIA" in controller:
                    nvidia_controllers.append(controller.strip())
                    
        if nvidia_controllers:
            print("\nNVIDIA USB 控制器:")
            for controller in nvidia_controllers:
                print(f"- {controller}")
                if "Type-C" in controller:
                    print("  * 支持 USB Type-C")
                    # Check for Thunderbolt support
                    try:
                        tb_check = subprocess.check_output(
                            ["wmic", "path", "Win32_PnPEntity", "where", 
                             "caption like '%Thunderbolt%'", "get", "caption"], 
                            encoding='gbk'
                        )
                        if len(tb_check.strip().split('\n')) > 1:  # Has content beyond header
                            print("  * 支持雷電接口")
                    except:
                        pass
        else:
            print("未找到 NVIDIA USB 控制器")
    except Exception as e:
        print(f"獲取 USB 控制器信息失敗: {e}")

if __name__ == "__main__":
    get_gpu_info()
    get_usb_controller_info()