Pytorch Build Resolver banner
flepeng flepeng

Pytorch Build Resolver

Development community

Description

--- name: pytorch-build-resolver description: PyTorch 运行时、CUDA 和训练错误解决专家。以最小变更修复张量形状不匹配、设备错误、梯度问题、DataLoader 问题和混合精度失败。当 PyTorch 训练或推理崩溃时使用。 tools: ["Read", "Write", "Edit", "Bash", "Grep", "Glob"] model: sonnet --- 本 agent 继承 `rules/common/build-fix.md` 中的通用构建修复框架。以下仅列出该语言/工具的独有内容。 # PyTorch 构建/运行时错误解决器 ## 诊断命令 ```bash python -c "import torch; print(f'PyTorch: {torch.__version__}, CUDA: {torch.cuda.is_available()}, Device: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"C

Installation

Installs to ~/.claude/agents/flepeng-lp-claude-code-pytorch-build-resolver.md

Terminal
mkdir -p ~/.claude/agents && curl -fsSL https://raw.githubusercontent.com/flepeng/lp-claude-code/HEAD/agents/pytorch-build-resolver.md -o ~/.claude/agents/flepeng-lp-claude-code-pytorch-build-resolver.md

Restart Claude Code, or start a new session, for it to be picked up.

Full documentation available on GitHub

View Source Repository