OpenVLA-OFT部署检查清单

硬件检查

• [ ] GPU内存 >= 16GB (推荐24GB+)
• [ ] 系统内存 >= 32GB
• [ ] 磁盘空间 >= 50GB (模型文件约25GB)

软件环境

• [ ] Python版本 3.8-3.10
• [ ] PyTorch版本 2.0.1+
• [ ] CUDA版本 11.7+
• [ ] 所有依赖项已安装 (pip list | grep -f requirements.txt)

模型文件

• [ ] 模型权重文件完整 (4个model-xxxx-of-00004.safetensors)
• [ ] 动作头文件存在 (action_head--150000_checkpoint.pt)
• [ ] 配置文件完整 (config.json, configuration_prismatic.py等)

功能验证

• [ ] 模型可成功加载 (无CUDA错误或权重不匹配)
• [ ] 示例图像可正常处理
• [ ] 动作可正常生成 (无NaN或异常值)
• [ ] 推理延迟在可接受范围 (<500ms/步)

### 5.5 问题诊断决策树

遇到部署问题时，可使用以下决策树进行诊断：

```mermaid
flowchart TD
    A[问题类型] --> B{导入错误}
    A --> C{运行时错误}
    A --> D{性能问题}
    
    B --> B1[检查依赖版本]
    B --> B2[检查Python路径]
    B --> B3[检查模型文件完整性]
    
    C --> C1{CUDA错误}
    C --> C2{维度不匹配}
    C --> C3{权重加载失败}
    
    C1 --> C1a[检查CUDA版本匹配性]
    C1 --> C1b[检查GPU内存是否充足]
    C1 --> C1c[尝试CPU模式验证代码逻辑]
    
    C2 --> C2a[检查输入图像尺寸]
    C2 --> C2b[检查批处理大小]
    C2 --> C2c[检查本体感知数据维度]
    
    D --> D1[启用量化减少内存使用]
    D --> D2[降低图像分辨率]
    D --> D3[优化批处理策略]