# @desc: PyTorch + CUDA 环境自检: 版本/GPU型号/显存
# @usage: wcl cudaprint.sh
#   不带参数, 直接跑; 需已装 PyTorch
python << 'PYEOF'
import torch
print(f'✓ PyTorch版本: {torch.__version__}')
print(f'✓ CUDA可用: {torch.cuda.is_available()}')
if torch.cuda.is_available():
    print(f'✓ CUDA版本: {torch.version.cuda}')
    print(f'✓ GPU: {torch.cuda.get_device_name(0)}')
    print(f'✓ 显存: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.1f} GB')
PYEOF