qing
63cfdd2066
- 第 1 轮安全审计修复: F-1~F-10（路径校验、预训练架构修正等） - 第 2 轮复审修复 11 项: 3 项上轮修复不完整（R01 checkpoint 键校验/ R02 fused_hidden_dim 值域/R03 pretrained=False）+ 8 项新问题 （actions.npy 形状校验、帧数警告、CLI 值域校验、stats 形状与路径、 onnxscript 依赖、归一化语义、解析去重） - 单元测试由 17 扩展至 51 用例（50 通过 + 1 环境性跳过） - 审计文档两轮合并至 actpred_CODE_AUDIT_20260808.md，更新 INDEX
18 lines
627 B
Bash
Executable File
18 lines
627 B
Bash
Executable File
#!/bin/bash
|
|
cd "$(dirname "$0")" || exit
|
|
python3 -m venv venv
|
|
source venv/bin/activate
|
|
|
|
# CPU-only PyTorch (no CUDA dependency)
|
|
# 如需 GPU 支持，注释掉下面一行并改用：pip install torch torchvision
|
|
pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
|
|
|
|
# Remaining deps from Tsinghua mirror
|
|
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
# ONNX 导出依赖（torch>=2.12 的 torch.onnx.export 走 dynamo 路径，onnxscript 为硬依赖）
|
|
pip install onnxscript onnxruntime -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
pip check
|
|
mkdir -p checkpoints data
|