AITemplate
AITemplate
📖 简介
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ 代码. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inferen
查看英文原版
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inferen
📥 安装此技能
ai-agent install aitemplate
📖 其他安装方式
方法 2：从 GitHub 克隆
git clone https://github.com/facebookincubator/AITemplatecd aitemplateai-agent link .
方法 3：手动安装
# 下载技能后复制到技能目录 cp -r aitemplate ~/.ai-agent/skills/
💡 提示： 技能将安装到你的本地 目录，不会存储在我们的服务器上。